sigs.k8s.io/kueue@v0.6.2/site/static/examples/jobs/sample-xgboostjob.yaml (about)

     1  apiVersion: kubeflow.org/v1
     2  kind: XGBoostJob
     3  metadata:
     4    name: xgboost-dist-iris-test-train
     5    namespace: default
     6    labels:
     7      kueue.x-k8s.io/queue-name: user-queue
     8  spec:
     9    xgbReplicaSpecs:
    10      Master:
    11        replicas: 1
    12        restartPolicy: Never
    13        template:
    14          spec:
    15            containers:
    16              - name: xgboost
    17                image: docker.io/kubeflow/xgboost-dist-iris:latest
    18                resources:
    19                  requests:
    20                    cpu: 0.5
    21                    memory: 256Mi
    22                ports:
    23                  - containerPort: 9991
    24                    name: xgboostjob-port
    25                imagePullPolicy: Always
    26                args:
    27                  - --job_type=Train
    28                  - --xgboost_parameter=objective:multi:softprob,num_class:3
    29                  - --n_estimators=10
    30                  - --learning_rate=0.1
    31                  - --model_path=/tmp/xgboost-model
    32                  - --model_storage_type=local
    33      Worker:
    34        replicas: 2
    35        restartPolicy: ExitCode
    36        template:
    37          spec:
    38            containers:
    39              - name: xgboost
    40                image: docker.io/kubeflow/xgboost-dist-iris:latest
    41                resources:
    42                  requests:
    43                    cpu: 0.5
    44                    memory: 256Mi
    45                ports:
    46                  - containerPort: 9991
    47                    name: xgboostjob-port
    48                imagePullPolicy: Always
    49                args:
    50                  - --job_type=Train
    51                  - --xgboost_parameter="objective:multi:softprob,num_class:3"
    52                  - --n_estimators=10
    53                  - --learning_rate=0.1