Skip to content

Instantly share code, notes, and snippets.

@sofianhamiti

sofianhamiti/step_train.py

Created March 22, 2022 13:17
Show Gist options
  • Save sofianhamiti/55d1481fbcf06c0efe167f1bfd011c82 to your computer and use it in GitHub Desktop.
Save sofianhamiti/55d1481fbcf06c0efe167f1bfd011c82 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
step_train.py
step_train = TrainingStep(
name="TrainAbaloneModel",
estimator=xgb_train,
inputs={...},
retry_policies=[
step_retry_policy,
job_retry_policy
]
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment