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

     1  apiVersion: kubeflow.org/v1
     2  kind: XGBoostJob
     3  metadata:
     4    name: lightgbm-dist-train-test
     5  spec:
     6    xgbReplicaSpecs:
     7      Master:
     8        replicas: 1
     9        restartPolicy: Never
    10        template:
    11          spec:
    12            containers:
    13            - name: xgboost
    14              image: kubeflow/lightgbm-dist-py-test:1.0
    15              ports:
    16              - containerPort: 9991
    17                name: xgboostjob-port
    18              imagePullPolicy: Never
    19              args:
    20                - --job_type=Train
    21                - --boosting_type=gbdt
    22                - --objective=binary
    23                - --metric=binary_logloss,auc
    24                - --metric_freq=1
    25                - --is_training_metric=true
    26                - --max_bin=255
    27                - --data=data/binary.train
    28                - --valid_data=data/binary.test
    29                - --num_trees=100
    30                - --learning_rate=01
    31                - --num_leaves=63
    32                - --tree_learner=feature
    33                - --feature_fraction=0.8
    34                - --bagging_freq=5
    35                - --bagging_fraction=0.8
    36                - --min_data_in_leaf=50
    37                - --min_sum_hessian_in_leaf=50
    38                - --is_enable_sparse=true
    39                - --use_two_round_loading=false
    40                - --is_save_binary_file=false
    41      Worker:
    42        replicas: 2
    43        restartPolicy: ExitCode
    44        template:
    45          spec:
    46            containers:
    47            - name: xgboost
    48              image: kubeflow/lightgbm-dist-py-test:1.0
    49              ports:
    50              - containerPort: 9991
    51                name: xgboostjob-port
    52              imagePullPolicy: Never
    53              args:
    54                - --job_type=Train
    55                - --boosting_type=gbdt
    56                - --objective=binary
    57                - --metric=binary_logloss,auc
    58                - --metric_freq=1
    59                - --is_training_metric=true
    60                - --max_bin=255
    61                - --data=data/binary.train
    62                - --valid_data=data/binary.test
    63                - --num_trees=100
    64                - --learning_rate=01
    65                - --num_leaves=63
    66                - --tree_learner=feature
    67                - --feature_fraction=0.8
    68                - --bagging_freq=5
    69                - --bagging_fraction=0.8
    70                - --min_data_in_leaf=50
    71                - --min_sum_hessian_in_leaf=50
    72                - --is_enable_sparse=true
    73                - --use_two_round_loading=false
    74                - --is_save_binary_file=false
    75