How to create a hyperparameter tuning step in SageMaker pipeline?

Question

I am trying to use the latest SageMaker Python SDK (v2.23.0) to implement a SageMaker pipeline that includes a hyperparameter tuning job. However I didn't see anything in module sagemaker.workflow.steps or sagemaker.workflow.step_collections that I can use. There is a TrainingStep class but it's not for HPO.

Is this not supported at this time?

score 1 · Answer 1 · edited Oct 14 '21 at 13:06

1

SageMaker Pipelines now supports a TuningStep to integrate Hyperparameter Tuning jobs in a Pipeline.

This example notebook shows how you can do this.

Source: AWS Employee

edited Oct 14 '21 at 13:06

Aakash Parsi

93
2
10

answered Jun 16 '21 at 21:32

Kirit Thadaka

429
2
5

How to create a hyperparameter tuning step in SageMaker pipeline?

1 Answers1