flytekitplugins.awssagemaker.AlgorithmName#

class flytekitplugins.awssagemaker.AlgorithmName[source]#

The algorithm name is used for deciding which pre-built image to point to. This is only required for use cases where SageMaker’s built-in algorithm mode is used. While we currently only support a subset of the algorithms, more will be added to the list. See: https://docs.aws.amazon.com/sagemaker/latest/dg/algos.html

Methods

Attributes

CUSTOM = 0#
XGBOOST = 1#