创建模型组
模型组包含一组受版本控制的模型。使用 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 中创建模型组,请完成以下步骤。
-
登录 Studio。有关更多信息,请参阅 加入 Amazon SageMaker 域。
-
在左侧导航窗格中,选择主页图标 ( )。
-
选择模型,然后选择模型注册表。
-
选择操作,然后选择创建模型组。
-
在创建模型组对话框中,输入以下信息:
-
在模型组的名称字段中输入新模型组的名称。
-
(可选)在描述字段中输入模型组的描述。
-
(可选)在标签字段中输入要与该模型组关联的任何键值对。有关使用标签的信息,请参阅《Amazon Web Services 一般参考》中的标记 Amazon 资源。
-
(可选)在项目字段中选择要与该模型组关联的项目。有关项目的信息,请参阅使用 SageMaker 项目自动执行 MLOps。
-
-
选择创建模型组。