Skip to content

Instantly share code, notes, and snippets.

@Geremie
Created December 16, 2020 09:55
Show Gist options
  • Save Geremie/8460b8f341e868dbc3178e5f712ac11c to your computer and use it in GitHub Desktop.
Save Geremie/8460b8f341e868dbc3178e5f712ac11c to your computer and use it in GitHub Desktop.
Automate your ML model retraining with Kubeflow
ML_ENGINE_TRAIN_OP = comp.load_component_from_url(
'https://raw.githubusercontent.com/kubeflow/pipelines/1.1.1-beta.1/components/gcp/ml_engine/train/component.yaml')
ml_engine_train_task = ML_ENGINE_TRAIN_OP(
project_id=project_id,
python_module=PYTHON_MODULE,
package_uris=package_uris,
region=REGION,
args=args,
job_dir=job_dir,
python_version=PYTHON_VERSION,
runtime_version=RUNTIME_VERSION,
job_id=job_id,
wait_interval=WAIT_INTERVAL)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment