AutoMLAlgorithmConfig - Amazon SageMaker

AutoMLAlgorithmConfig

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

Contents

AutoMLAlgorithms

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

Note

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.

  • In ENSEMBLING mode:

    • "catboost"

    • "extra-trees"

    • "fastai"

    • "lightgbm"

    • "linear-learner"

    • "nn-torch"

    • "randomforest"

    • "xgboost"

  • In HYPERPARAMETER_TUNING mode:

    • "linear-learner"

    • "mlp"

    • "xgboost"

Type: Array of strings

Array Members: Maximum number of 11 items.

Valid Values: xgboost | linear-learner | mlp | lightgbm | catboost | randomforest | extra-trees | nn-torch | fastai

Required: Yes

See Also

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