CODEWITHSUNDEEP
amazon-sagemaker
Ori N
Ori N

Reputation: 748

Available framework version for AWS SageMaker SKLearn

I'm using SageMaker Python SDK's SKLearn class which has a parameter called framework version. Its default is 0.20.0. I want to use a different version of Scikit-learn (0.21.0 or higher). It is possible? How do I know which versions are supported?

Upvotes: 4

Views: 3271

Answers (1)

Liam Roberts
Liam Roberts

Reputation: 1281

I actually couldn't find a nice link that showed this, but if you enter an incorrect framework the error message will print out the available versions.

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.

Upvotes: 4

Related Questions