Skip to content

Instantly share code, notes, and snippets.

@post2web

post2web/xgboost docker training.md

Last active January 30, 2020 23:50
Show Gist options
  • Save post2web/1ff4a3b720664c67a6d25bc6be22a07f to your computer and use it in GitHub Desktop.
Save post2web/1ff4a3b720664c67a6d25bc6be22a07f to your computer and use it in GitHub Desktop.
Download ZIP
Local training of ML containers
Raw
xgboost docker training.md 
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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment