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://github.com/aws/sagemaker-python-sdk#scikit-learn-sagemaker-estimators.

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.