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
Last active
January 29, 2020 20:07
-
-
Save post2web/3e066dbc8f9639cfc08e83b0ba9569bf to your computer and use it in GitHub Desktop.
Training ML containers on the Cloud
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment