View a markdown version of this page

Scikit-learn 与 Amazon A SageMaker I 配合使用的资源 - 亚马逊 SageMaker AI

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

Scikit-learn 与 Amazon A SageMaker I 配合使用的资源

您可以使用 Amazon SageMaker AI 使用自定义 Scikit-learn 代码训练和部署模型。 SageMaker AI Python SDK Scikit-learn 估算器和模型以及 SageMaker AI 开源 Scikit-learn 容器使编写 Scikit-learn 脚本和在 A SageMaker I 中运行脚本变得更加容易。以下部分提供了参考资料,您可以用来学习如何 Scikit-learn 与 SageMaker AI 配合使用。

要求

Scikit-learn 1.4 具有以下依赖关系。

依赖关系 最低版本
Python 3.10
NumPy 2.1.0
SciPy 1.15.3
joblib 1.5.2
threadpoolctl 3.6.0

A SageMaker I Scikit-learn 容器支持以下 Scikit-learn 版本。

支持的 Scikit-learn 版本 最低 Python 版本
1.4-2 3.10
1.2-1 3.8
1.0-1 3.7
0.23-1 3.6
0.20.0 2.73.4

有关编写 Scikit-learn 训练脚本以及在 SageMaker AI 中使用 Scikit-learn 估算器和模型的一般信息,请参阅与 Pyth SageMaker on SDK Scikit-learn 配合使用

您需要做什么?

注意

运行 AI 示例笔记本需要 Matplotlib v2.2.3 或更高版本。 SageMaker Scikit-learn

我想用 Scikit-learn 于 SageMaker AI 中的数据处理、特征工程或模型评估。

有关 Jupyter 笔记本的示例,请参阅。https://github.com/awslabs/amazon-sagemaker-examples/tree/master/sagemaker_processing/scikit_learn_data_processing_and_model_evaluation

有关训练和部署 Scikit-learn 模型的博客文章,请参阅 Amazon SageMaker AI 添加 Scikit-Learn 支持

有关文档,请参阅 ReadTheDocs

我想在 SageMaker AI 中训练一个自定义 Scikit-learn 模型。

有关 Jupyter 笔记本的示例,请参阅。https://github.com/awslabs/amazon-sagemaker-examples/tree/master/sagemaker-python-sdk/scikit_learn_iris

有关文档,请参阅使用训练模型 Scikit-learn

我有一个在 SageMaker AI 中训练过的 Scikit-learn 模型,我想将其部署到托管端点。

有关更多信息,请参阅部署 Scikit-learn 模型

我有一个在 SageMaker AI 之外训练的 Scikit-learn 模型,我想将其部署到 A SageMaker I 端点

有关更多信息,请参阅从模型数据部署端点

我想查看亚马逊 SageMaker Python 软件开发工具包 Scikit-learn 类的 API 文档。

有关更多信息,请参阅Scikit-learn类

我想查看有关 SageMaker AI Scikit-learn 容器的信息。

有关更多信息,请参阅SageMaker Scikit-learn 容器 GitHub 存储库