docker run \
-v "${PWD}":/local-drive \
-it gcr.io/aihub-c2t-containers/kfp-components/trainer/dist_xgboost:latest \
--output-location /local-drive/xgboost-output \
--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 30, 2020 23:50
-
-
Save post2web/1ff4a3b720664c67a6d25bc6be22a07f to your computer and use it in GitHub Desktop.
Local training of ML containers
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment