Skip to content

Instantly share code, notes, and snippets.

@post2web

post2web/xgboost ai-platform training.md

Last active January 29, 2020 20:07
Show Gist options
  • Save post2web/3e066dbc8f9639cfc08e83b0ba9569bf to your computer and use it in GitHub Desktop.
Save post2web/3e066dbc8f9639cfc08e83b0ba9569bf to your computer and use it in GitHub Desktop.
Download ZIP
Training ML containers on the Cloud
Raw
xgboost ai-platform training.md 
gcloud ai-platform jobs submit training my_xgboost_job \
  --stream-logs \
  --scale-tier CUSTOM \
  --master-machine-type n1-standard-8 \
  --master-image-uri gcr.io/aihub-c2t-containers/kfp-components/trainer/dist_xgboost:latest \
  -- \
  --output-location gs://REPLACE_WITH_YOUR_OUTPUT_BUCKET \
  --training-data gs://aihub-c2t-containers-public/release-0.2.0/kfp-components/trainer/dist_xgboost/data/csv/iris/train.csv \
  --validation-data gs://aihub-c2t-containers-public/release-0.2.0/kfp-components/trainer/dist_xgboost/data/csv/iris/valid.csv \
  --target-column target \
  --data-type csv \
  --number-of-classes 3 \
  --objective multi:softprob
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment