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

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

创建模型组

模型组包含一组受版本控制的模型。使用 Amazon SDK for Python (Boto3) 或 Amazon SageMaker Studio 控制台创建模型组。

创建模型组 (Boto3)

重要

允许 Amazon SageMaker Studio 或 Amazon SageMaker Studio Classic 创建亚马逊 SageMaker资源的自定义 IAM 策略还必须授予向这些资源添加标签的权限。需要向资源添加标签的权限,因为 Studio 和 Studio Classic 会自动标记他们创建的任何资源。如果 IAM 策略允许 Studio 和 Studio Classic 创建资源但不允许标记,则在尝试创建资源时可能会出现 AccessDenied “” 错误。有关更多信息,请参阅 提供为资源添加标签 SageMaker的权限

Amazon 适用于亚马逊的托管政策 SageMaker授予创建 SageMaker 资源的权限已经包括在创建这些资源时添加标签的权限。

要使用 Boto3 创建模型组,请调用 create_model_package_group API 操作并指定名称和描述作为参数。以下示例说明了如何创建模型组。create_model_package_group 调用的响应是新模型组的 Amazon 资源名称 (ARN)。

首先,导入所需的软件包并设置 SageMaker Boto3 客户端。

import time import os from sagemaker import get_execution_role, session import boto3 region = boto3.Session().region_name role = get_execution_role() sm_client = boto3.client('sagemaker', region_name=region)

现在创建模型组。

import time model_package_group_name = "scikit-iris-detector-" + str(round(time.time())) model_package_group_input_dict = { "ModelPackageGroupName" : model_package_group_name, "ModelPackageGroupDescription" : "Sample model package group" } create_model_package_group_response = sm_client.create_model_package_group(**model_package_group_input_dict) print('ModelPackageGroup Arn : {}'.format(create_model_package_group_response['ModelPackageGroupArn']))

创建模型组(工作室或经典工作室)

要在 Amazon SageMaker Studio 控制台中创建模型组,请根据您使用的是 Studio 还是 Studio Classic 完成以下步骤。

Studio
  1. 按照启动 Amazon SageMaker Studio 中的说明打开 SageMaker Studio 控制台。

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

  3. 如果尚未选择,请选择已注册的模型选项卡。

  4. 已注册模型选项卡标签的正下方,选择模型组(如果尚未选择)。

  5. 选择注册,然后选择模型组

  6. 注册模型组对话框中,输入以下信息:

    • 模型组名称字段中新模型组的名称

    • (可选)“描述” 字段中对模型组的描述

    • (可选)您要在标签字段中与模型组关联的任何键值对。有关使用标签的信息,请参阅《Amazon Web Services 一般参考》中的标记 Amazon 资源

  7. 选择注册模型组

  8. (可选)在模型页面中,选择已注册的模型选项卡,然后选择模型组。确认新创建的模型组出现在模型组列表中。

Studio Classic
  1. 登录亚马逊 SageMaker Studio 经典版。有关更多信息,请参阅启动 Amazon SageMaker Studio 经典版

  2. 在左侧导航窗格中,选择主页图标 ( )。

  3. 选择模型,然后选择模型注册表

  4. 选择操作,然后选择创建模型组

  5. 创建模型组对话框中,输入以下信息:

    • 模型组的名称字段中输入新模型组的名称。

    • (可选)在描述字段中输入模型组的描述。

    • (可选)在标签字段中输入要与该模型组关联的任何键值对。有关使用标签的信息,请参阅《Amazon Web Services 一般参考》中的标记 Amazon 资源

    • (可选)在项目字段中选择要与该模型组关联的项目。有关项目的信息,请参阅使用项目自动执行 MLOP SageMaker

  6. 选择创建模型组