Class: Aws::SageMaker::Types::AutoMLAlgorithmConfig

Inherits:
Struct
  • Object
show all
Includes:
Aws::Structure
Defined in:
lib/aws-sdk-sagemaker/types.rb

Overview

The collection of algorithms run on a dataset for training the model candidates of an Autopilot job.

Constant Summary collapse

SENSITIVE =
[]

Instance Attribute Summary collapse

Instance Attribute Details

#auto_ml_algorithmsArray<String>

The selection of algorithms run on a dataset to train the model candidates of an Autopilot job.

<note markdown=“1”> Selected algorithms must belong to the list corresponding to the training mode set in [AutoMLJobConfig.Mode] (‘ENSEMBLING` or `HYPERPARAMETER_TUNING`). Choose a minimum of 1 algorithm.

</note>
  • In ‘ENSEMBLING` mode:

    • “catboost”

    • “extra-trees”

    • “fastai”

    • “lightgbm”

    • “linear-learner”

    • “nn-torch”

    • “randomforest”

    • “xgboost”

  • In ‘HYPERPARAMETER_TUNING` mode:

    • “linear-learner”

    • “mlp”

    • “xgboost”

[1]: docs.aws.amazon.com/sagemaker/latest/APIReference/API_AutoMLJobConfig.html#sagemaker-Type-AutoMLJobConfig-Mode

Returns:

  • (Array<String>)


1709
1710
1711
1712
1713
# File 'lib/aws-sdk-sagemaker/types.rb', line 1709

class AutoMLAlgorithmConfig < Struct.new(
  :auto_ml_algorithms)
  SENSITIVE = []
  include Aws::Structure
end