Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save Sandy4321/8271758b654645512e63bf8c70ab0136 to your computer and use it in GitHub Desktop.
Save Sandy4321/8271758b654645512e63bf8c70ab0136 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