HyperParameterTuningJobConfig - Amazon SageMaker

HyperParameterTuningJobConfig

Configures a hyperparameter tuning job.

Contents

ResourceLimits

The ResourceLimits object that specifies the maximum number of training and parallel training jobs that can be used for this hyperparameter tuning job.

Type: ResourceLimits object

Required: Yes

Strategy

Specifies how hyperparameter tuning chooses the combinations of hyperparameter values to use for the training job it launches. For information about search strategies, see How Hyperparameter Tuning Works.

Type: String

Valid Values: Bayesian | Random | Hyperband | Grid

Required: Yes

HyperParameterTuningJobObjective

The HyperParameterTuningJobObjective specifies the objective metric used to evaluate the performance of training jobs launched by this tuning job.

Type: HyperParameterTuningJobObjective object

Required: No

ParameterRanges

The ParameterRanges object that specifies the ranges of hyperparameters that this tuning job searches over to find the optimal configuration for the highest model performance against your chosen objective metric.

Type: ParameterRanges object

Required: No

RandomSeed

A value used to initialize a pseudo-random number generator. Setting a random seed and using the same seed later for the same tuning job will allow hyperparameter optimization to find more a consistent hyperparameter configuration between the two runs.

Type: Integer

Valid Range: Minimum value of 0.

Required: No

StrategyConfig

The configuration for the Hyperband optimization strategy. This parameter should be provided only if Hyperband is selected as the strategy for HyperParameterTuningJobConfig.

Type: HyperParameterTuningJobStrategyConfig object

Required: No

TrainingJobEarlyStoppingType

Specifies whether to use early stopping for training jobs launched by the hyperparameter tuning job. Because the Hyperband strategy has its own advanced internal early stopping mechanism, TrainingJobEarlyStoppingType must be OFF to use Hyperband. This parameter can take on one of the following values (the default value is OFF):

OFF

Training jobs launched by the hyperparameter tuning job do not use early stopping.

AUTO

SageMaker stops training jobs launched by the hyperparameter tuning job when they are unlikely to perform better than previously completed training jobs. For more information, see Stop Training Jobs Early.

Type: String

Valid Values: Off | Auto

Required: No

TuningJobCompletionCriteria

The tuning job's completion criteria.

Type: TuningJobCompletionCriteria object

Required: No

See Also

For more information about using this API in one of the language-specific AWS SDKs, see the following: