sagemaker/aws.sdk.kotlin.services.sagemaker.model/AutoMlAlgorithmConfig/Builder/autoMlAlgorithms

autoMlAlgorithms

var autoMlAlgorithms: List<AutoMlAlgorithm>?

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

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"