发布于: Apr 25, 2023

Amazon SageMaker Python SDK 是一个开源库,可用于在 Amazon SageMaker 上训练和部署机器学习模型。我们很高兴地宣布,SageMaker Python SDK 现在可以帮助数据科学家在只需最少代码更改的情况下,将在其首选 IDE 和本地笔记本中编写的任何本地机器学习代码以及相关的运行时系统依赖项作为大规模机器学习模型训练作业执行。

数据科学家只需要在本地机器学习代码中添加一行代码(Python 装饰器),SageMaker Python SDK 会将他们的代码、数据集和工作空间环境设置作为 SageMaker 训练作业运行。这种装饰器模式可以减少对自定义代码构造和环境变量管理的需求,从而帮助数据科学家更轻松地在 SageMaker 上启动机器学习工作流程。此外,这种对本地代码转换为作业的增强体验通过自动捕获和复制本地运行时系统减少了在容器管理上花费的时间,使数据科学家可以在生产级作业中花费更少的时间重新创建本地环境。 

此功能现已在提供 Amazon SageMaker Python SDK 的所有区域推出。要开始使用这项新功能,请阅读 Amazon SageMaker 文档Amazon SageMaker Python SDK 指南SageMaker 模型训练页面。 

要查看示例并详细了解如何使用本地代码转换为训练作业 SDK 接口,请访问机器学习博客并查看 SageMaker Python SDK 存储库中的示例笔记本