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

创建模型组

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

创建模型组 (Boto3)

要使用 Boto3 创建模型组,请调用 create_model_package_group 方法并指定名称和描述作为参数。以下示例说明了如何创建模型组。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)

要在 Amazon SageMaker Studio 中创建模型组,请完成以下步骤。

  1. 登录 Studio。有关更多信息,请参阅 加入 Amazon SageMaker 域

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

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

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

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

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

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

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

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

  6. 选择创建模型组