github.com/kubeflow/training-operator@v1.7.0/examples/xgboost/xgboost-dist/xgboostjob_v1alpha1_iris_train.yaml (about)

     1  apiVersion: kubeflow.org/v1
     2  kind: XGBoostJob
     3  metadata:
     4    name: xgboost-dist-iris-test-train
     5  spec:
     6    xgbReplicaSpecs:
     7      Master:
     8        replicas: 1
     9        restartPolicy: Never
    10        template:
    11          apiVersion: v1
    12          kind: Pod
    13          spec:
    14            containers:
    15            - name: xgboost
    16              image: docker.io/merlintang/xgboost-dist-iris:1.1
    17              ports:
    18              - containerPort: 9991
    19                name: xgboostjob-port
    20              imagePullPolicy: Always
    21              args:
    22                - --job_type=Train
    23                - --xgboost_parameter=objective:multi:softprob,num_class:3
    24                - --n_estimators=10
    25                - --learning_rate=0.1
    26                - --model_path=autoAI/xgb-opt/2
    27                - --model_storage_type=oss
    28                - --oss_param=unknown
    29      Worker:
    30        replicas: 2
    31        restartPolicy: ExitCode
    32        template:
    33          apiVersion: v1
    34          kind: Pod
    35          spec:
    36            containers:
    37            - name: xgboost
    38              image: docker.io/merlintang/xgboost-dist-iris:1.1
    39              ports:
    40              - containerPort: 9991
    41                name: xgboostjob-port
    42              imagePullPolicy: Always
    43              args:
    44                - --job_type=Train
    45                - --xgboost_parameter="objective:multi:softprob,num_class:3"
    46                - --n_estimators=10
    47                - --learning_rate=0.1
    48  
    49