Amazon SageMaker Model Registry / Pipelines - 如何为给定的 Model 版本手动设置 Stage？

Question

这可能是一个非常具体的问题，但无论如何我都会尝试。

我想为给定的 Model 版本显式设置 Model 注册表中的 Stage 列：

此图片来自文档，只有在您运行他们提供的示例 SageMaker Projects MLOps 模板时才会设置。 当我手动创建Model Package （即Model Version ）时，该列仍然为空。 我该如何设置？ API 叫什么？

另外浏览model版本历史的文档中有如下一句话

我们如何手动发送那个确切的事件（“部署到阶段 XYZ”）？

我已经彻底检查了 SageMaker MLOps 项目生成的所有文件（CodeBuild Builds、CodePipeline、CloudFormation、各种.py文件、SageMaker Pipeline），但找不到对该事件的任何直接和显式调用。

我认为它可能以某种方式连接到标签sagemaker:deployment-stage但我已经将它设置在Endpoint 、 EndpointConfiguration和Model上，但没有成功。 我还尝试盲目调用UpdateModelPackage API 并在CustomerMetadataProperties中设置Stage 。 再次 - 没有运气。

我在该活动选项卡中得到的唯一信息是给定的 Model 版本已部署到推理端点：

Answer 1

我遇到了完全相同的问题，我想更改 model 阶段，但在 AWS 提供的示例代码中找不到它在哪里完成。

经过一些研究并查看示例代码后，我意识到它是在云形成执行中完成的。 首先他们添加标签

'sagemaker:deployment-stage': stage_config['Parameters']['StageName']

然后云形成执行（cfnUpdate 调用）更新阶段并部署。

我找不到另一种方法来通过调用 update_model_package 或其他方法来更改 state。

Answer 2

您可以使用create_model_package API 或update_model_package API 中的ModelApprovalStatus参数设置状态

Model package state change should create an event in EventBridge (like many other SageMaker events) https://docs.aws.amazon.com/sagemaker/latest/dg/automating-sagemaker-with-eventbridge.html#eventbridge-model-package ，这使您能够运行您选择的自动化。

在默认的 SageMaker 管道项目模板中，您可以在为部署创建的 CodePipeline 管道中看到 EventBridge 驱动的建议逻辑：您可以在顶部看到“触发器 - CloudWatchEvent”。 您不会在 git 中将事件源视为代码，因为预计状态更改将在该演示模板的 Studio model 注册表 UI 中完成。

Model Registry 发出的那些 EventBridge 事件也可以在一些博客中看到：

• 使用 MLOps 在 AWS 上驯服机器学习：参考架构

• 多账户、中心辐射型 Amazon SageMaker model 注册表的模式

• 使用 Amazon SageMaker 项目、GitLab 和 GitLab 管道构建 MLOps 工作流程