flytekitplugins.awssagemaker.HPOJob#

class flytekitplugins.awssagemaker.HPOJob(max_number_of_training_jobs, max_parallel_training_jobs, tunable_params=None)[source]#

HPOJob Configuration should be used to configure the HPO Job.

Parameters

  • max_number_of_training_jobs (int) – maximum number of jobs to run for a training round

  • max_parallel_training_jobs (int) – limits the concurrency of the training jobs

  • tunable_params (Optional[List[str]]) – [optional] should be a list of parameters for which we want to provide the tuning ranges

Return type

None

Methods

Attributes

tunable_params: Optional[List[str]] = None
max_number_of_training_jobs: int
max_parallel_training_jobs: int