Amazon SageMaker
Developer Guide

Use Scikit-learn with Amazon SageMaker

You can use Amazon SageMaker to train and deploy a model using custom Scikit-learn code. The Amazon SageMaker Python SDK Scikit-learn estimators and models and the Amazon SageMaker open-source Scikit-learn container make writing a Scikit-learn script and running it in Amazon SageMaker easier. For information about writing Scikit-learn training scripts and using Scikit-learn estimators and models with Amazon SageMaker, see https://sagemaker.readthedocs.io/en/stable/using_sklearn.html.

For API reference for the Amazon SageMaker Python SDK Scikit-learn classes, see https://sagemaker.readthedocs.io/en/stable/sagemaker.sklearn.html.

For information about Scikit-learn versions supported by the Amazon SageMaker Scikit-learn container, see https://github.com/aws/sagemaker-python-sdk/blob/master/src/sagemaker/sklearn/README.rst.

For information about how to build and contribute to the Amazon SageMaker Scikit-learn container, see the GitHub repository at https://github.com/aws/sagemaker-scikit-learn-container.