Skip to content

Instantly share code, notes, and snippets.

@rodrigofp-cit
Created December 24, 2019 13:22
Show Gist options
  • Save rodrigofp-cit/6eb4a02a79b6833c14d6173d070a0a82 to your computer and use it in GitHub Desktop.
Save rodrigofp-cit/6eb4a02a79b6833c14d6173d070a0a82 to your computer and use it in GitHub Desktop.
PROJECT_ID=<PROJECT_ID>
BUCKET_NAME="${PROJECT_ID}-black-friday-demo-bucket"
MODEL_NAME=black_friday_forecast
REGION=us-central1
VERSION='v'`date '+%Y%m%d_%H%M%S'`
MODEL_PATH=gs://$BUCKET_NAME/models/$VERSION/model
gcloud ai-platform jobs submit training black_friday_forecast_$(date +"%Y%m%d_%H%M%S") \
--job-dir gs://$BUCKET_NAME/models/$VERSION/job_dir \
--package-path ./train \
--module-name train.train \
--region $REGION \
--runtime-version 1.15 \
--python-version 2.7 \
--scale-tier BASIC \
--stream-logs \
--project ${PROJECT_ID} \
-- \
--csv-data-path gs://${BUCKET_NAME}/data/train.csv \
--model-dir $MODEL_PATH \
--model-name random_forest
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment