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

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

使用模型注册表注册并部署模型

使用 SageMaker 模型注册表您可以执行以下操作:

  • 用于生产的目录模型。

  • 管理模型版本。

  • 将元数据(例如训练指标)与模型关联。

  • 管理模型的批准状态。

  • 将模型部署到生产中。

  • 使用 CI/CD 自动化模型部署。

通过创建包含模型不同版本的模型包组对模型进行编目。您可以创建一个模型组来跟踪为解决特定问题而训练的所有模型。然后,您可以注册训练的每个模型,模型注册表将其作为新模型版本添加到模型组中。典型的工作流程可能如下所示:

  • 创建模型组。

  • 创建训练模型的 ML 管道。有关的信息 SageMaker 管道,请参阅创建和管理 SageMaker 管道.

  • 对于 ML 管道的每次运行,请创建一个模型版本,您在第一步中创建的模型组中注册该模型版本。

模型注册表结构

这些区域有: SageMaker 模型注册表结构为几个模型组,每个组中都有模型包。模型组中的每个模型包对应于训练的模型。每个模型包的版本是一个从 1 开始的数值,随着每个新模型包添加到模型组中而递增。例如,如果将 5 个模型包添加到模型组中,则模型包的版本将为 1、2、3、4 和 5。下图中显示的模型注册表示例包含 3 个模型组,其中每个组包含与特定机器学习问题相关的模型包。

SageMaker 中有两种类型的模型包。一种类型用于AmazonMarketplace,另一个在模型注册表中使用。中使用的模型包AmazonMarketplace 不是可版本的实体,也不与模型注册表中的模型组关联。有关中使用的模型包的更多信息Amazon请参阅 Marketplace在中购买和出售 Amazon SageMaker 算法和模型Amazon Web Services Marketplace.

模型注册表中使用的模型包是版本化的,并且必须与模型组关联。此模型包类型的 ARN 具有以下结构:'arn:aws:sagemaker:region:account:model-group/version'

以下主题向您介绍如何使用模型注册表。