具有自動駕駛步驟的 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 的集成訓練模式作為原生管道步驟，您可以參考公共文檔和示例筆記本。