wehub-resource-sync
14 lines
403 B
Plaintext
14 lines
403 B
Plaintext
name: xgboost-example
|
|
python_env: python_env.yaml
|
|
entry_points:
|
|
main:
|
|
parameters:
|
|
learning_rate: {type: float, default: 0.3}
|
|
colsample_bytree: {type: float, default: 1.0}
|
|
subsample: {type: float, default: 1.0}
|
|
command: |
|
|
python train.py \
|
|
--learning-rate={learning_rate} \
|
|
--colsample-bytree={colsample_bytree} \
|
|
--subsample={subsample}
|