在模型注册表中注册 SageMaker 模型版本 - Amazon SageMaker
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)

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

在模型注册表中注册 SageMaker 模型版本

使用 SageMaker Canvas,您可以构建模型的多个迭代或版本,以便随着时间的推移对其进行改进。如果您获得了更好的训练数据,或者您想尝试提高模型的准确性,您可能想要构建模型的新版本。有关向模型添加版本的更多信息,请参阅更新模型

在您构建模型并对模型感到满意之后,您可能需要对其性能进行评估,并在将其用于生产之前,让组织内的数据科学家或 MLOps 工程师对其进行审核。为此,您可以将模型版本注册到SageMaker 模型注册表。 SageMaker 模型注册表是一个存储库,数据科学家或工程师可以使用该存储库对机器学习 (ML) 模型进行编目并管理模型版本及其相关元数据,例如训练指标。他们还可以管理和记录模型的审批状态。

将模型版本注册到模型注册表后,数据科学家或您的 MLOPS 团队可以通过 SageMaker Studio Classic 访问 SageMaker 模型注册表,Studio Clas sic 是一个用于处理机器学习模型的基于 Web 的集成开发环境 (IDE)。 SageMaker 在 Studio Classic 的 SageMaker 模型注册表界面中,数据科学家或 MLOPS 团队可以评估您的模型并更新其批准状态。如果模型的性能不符合他们的要求,数据科学家或 MLOps 团队可以将状态更新为Rejected。如果模型的性能符合他们的要求,数据科学家或 MLOps 团队可以将状态更新为Approved。然后,他们可以将模型部署到端点,或使用 CI/CD 管道自动部署模型。您可以使用 SageMaker 模型注册表功能将在 Canvas 中构建的模型与组织中的 mLOPs 流程无缝集成。

下图总结了将在 Canvas 中内置的模型版本注册到 SageMaker 模型注册表以集成到 mLOps 工作流程中的示例。

Canvas 用户构建和注册模型版本、数据科学家或 MLOps 团队审核模型版本、自动工作流将版本部署到生产环境中的四个步骤示意图。

您可以将表格、图像和文本模型版本注册到 SageMaker 模型注册表。这包括时间序列预测模型和 JumpStart 基于微调的基础模型

注意

目前,您无法将 BYOM 模型版本或基于 Amazon Bedrock 的 Canvas 中内置的微调基础模型注册到模型注册表中。 SageMaker

以下各节介绍如何从 Canvas 将模型版本注册到 SageMaker 模型注册表。

权限管理

默认情况下,您有权将模型版本注册到 SageMaker 模型注册表。 SageMaker 通过策略为所有新的和现有的 Canvas 用户配置文件授予这些权限,该AmazonSageMakerCanvasFullAccess策略附加到托管 Canvas 应用程序的 SageMaker 域的 Amazon IAM 执行角色。

如果您的 Canvas 管理员正在设置新的域名或用户配置文件,则在设置域名并按照入门指南中的先决条件说明进行操作时, SageMaker 会通过 ML Ops 权限配置选项开启模型注册权限,该选项默认处于启用状态。

Canvas 管理员还可以在用户配置文件级别管理模型注册权限。例如,如果管理员希望向某些用户配置文件授予模型注册权限,但删除其他用户配置文件的权限,他们可以编辑特定用户的权限。以下过程说明如何关闭特定用户配置文件的模型注册权限:

  1. 打开 SageMaker 控制台,网址为 https://console.aws.amazon.com/sagemaker/

  2. 在左侧导航窗格中,选择管理员配置

  3. 管理员配置下,选择

  4. 从域名列表中,选择用户配置文件的域。

  5. 域名详细信息页面上,选择要编辑其权限的用户个人资料

  6. 用户详细信息页面上,选择编辑

  7. 在左侧导航窗格中,选择 Canvas 设置

  8. ML Ops 权限配置部分,关闭启用模型注册表注册权限开关。

  9. 选择提交以保存对您的域名设置的更改。

用户配置文件不应再拥有模型注册权限。

将模型版本注册到 SageMaker 模型注册表

SageMaker 模型注册表会跟踪您为解决模型组中的特定问题而构建的所有模型版本。当您构建 SageMaker Canvas 模型并将其注册到 SageMaker 模型注册表时,它会作为新的模型版本添加到模型组中。例如,如果您构建并注册了模型的四个版本,则在模型注册表界面中工作的数据科学家或 MLOPs 团队可以在一个位置查看模型组并查看模型的所有四个版本。 SageMaker

将 Canvas 模型注册到 SageMaker 模型注册表时,将自动创建一个模型组,并以您的 Canvas 模型命名。或者,您可以将其重命名为自己选择的名称,或者使用模型注册表中的现有 SageMaker 模型组。有关创建模型组的更多信息,请参阅创建模型组

注意

目前,您只能将在 Canvas 中构建的 SageMaker 模型注册到同一个账户的模型注册表中。

要从 Canvas 应用程序将 SageMaker 模型版本注册到模型注册表,请按以下步骤操作:

  1. 打开 SageMaker 画布应用程序。

  2. 在左侧导航窗格中,选择我的模型

  3. 我的模型页面上,选择您的模型。您可以按问题类型进行筛选,以便更轻松地找到您的模型。

  4. 选择模型后,将打开版本页面,其中列出了您的模型的所有版本。您可以打开显示高级指标开关来查看高级指标(如召回率精度),以比较您的模型版本并确定要注册哪个版本。

  5. 在模型版本列表中,对于要注册的版本,选择更多选项图标 ( )。或者,您可以双击需要注册的版本,然后在版本详细信息页面上,选择更多选项图标 ( )。

  6. 在下拉列表中,选择添加到模型注册表。此时将打开添加到模型注册表对话框。

  7. 添加到模型注册表对话框中,执行以下操作:

    1. (可选)在 SageMaker Studio Classic 模型组部分的模型组名称字段中,输入要向其注册版本的模型组的名称。您可以为为您 SageMaker 创建的新模型组指定名称,也可以指定现有模型组。如果不指定此字段,Canvas 会将您的版本注册到与模型同名的默认模型组中。

    2. 选择添加

现在,您的模型版本应该已注册到模型注册表中的 SageMaker 模型组。当您在模型注册表中将模型版本注册到 SageMaker 模型组时,Canvas 模型的所有后续版本都将注册到同一个模型组(如果您选择注册它们)。如果您将版本注册到其他模型组,则需要进入 SageMaker 模型注册表并删除该模型组。然后,您可以将模型版本重新注册到新的模型组中。

要查看模型的状态,可以返回 Canvas 应用程序中模型的版本页面。此页面显示每个版本的模型注册表状态。如果状态为Registered,则表示模型已成功注册。

如果要查看已注册模型版本的详细信息,对于模型注册表状态,可以将鼠标悬停在已注册字段上以查看模型注册表详细信息弹出框。这些详细信息包含更多信息,例如:

  • 模型包组名称是您的版本在模型注册表中注册到的 SageMaker 模型组。

  • 审批状态可以是Pending ApprovalApprovedRejected。如果 Studio Classic 用户在 SageMaker 模型注册表中批准或拒绝您的版本,则当您刷新页面时,模型版本页面上的此状态会更新。

以下屏幕截图显示了模型注册表详细信息框,以及该特定模型版本的Approved审批状态

Canvas 应用程序中 SageMaker 模型注册表详细信息框的屏幕截图。