Question

我正在使用SageMaker Python SDK的SKLearn类，该类具有一个称为框架版本的参数。其默认值为0.20.0。我想使用其他版本的Scikit-learn（0.21.0或更高版本）。有可能的？我怎么知道支持哪些版本？

Answer 1

我实际上找不到一个很好的链接来显示这一点，但是如果您输入了不正确的框架，错误消息将打印出可用版本。

estimator = SKLearn(
    entry_point = "train.py",
    source_dir = source_dir,
    instance_type='ml.m4.xlarge',
    role = role,
    framework_version='0.30.0',
    py_version = 'py3'
)

ValueError: Unsupported sklearn version: 0.30.0. You may need to upgrade your SDK version (pip install -U sagemaker) for newer sklearn versions. 
Supported sklearn version(s): 0.20.0, 0.23-1.

适用于AWS SageMaker SKLearn的可用框架版本

1 个答案: