具有自动驾驶步骤的 Sagemaker 管道

Question

我希望使用 Sagemaker 将 Autopilot AutoML 训练和部署步骤结合到更广泛的管道中。

我在有关将 AutoML 与管道一起使用的文档中找不到任何内容。

有没有人有在 Sagemaker 上尝试这样的事情的经验？

Answer 1

我不相信管道支持 AutoML，您可以在此处查看支持的步骤列表 - https://docs.aws.amazon.com/sagemaker/latest/dg/build-and-manage-steps.html

您可能会添加 Lambda 步骤并从 lambda 函数触发 AutoML 作业。 我还建议重新考虑流程，因为 AutoML 使用多种算法运行和训练。 虽然它非常适合探索，但它可能不是可重用管道的最佳选择。

Answer 2

目前，管道不支持 AutoML 步骤。 查看这篇博文，其中展示了如何使用 Lambdas 将 AutoML 集成到您的管道中。

https://aws.amazon.com/blogs/machine-learning/move-amazon-sagemaker-autopilot-ml-models-from-experimentation-to-production-using-amazon-sagemaker-pipelines/

特别是如何在 Lambda function 中启动 AutoML 的示例如下：

https://github.com/aws/amazon-sagemaker-examples/blob/main/autopilot/sagemaker-autopilot-pipelines/start_autopilot_job.py

Answer 3

您现在可以按照此示例将Sagemaker autopilot 步骤集成到您的管道中：

祝你好运

Answer 4

Sagemaker 管道现在仅支持 AutoML 的集成训练模式作为原生管道步骤，您可以参考公共文档和示例笔记本。