建立模型群組 - Amazon SageMaker

本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。

建立模型群組

模型群組包含不同版本的模型。您可以建立模型群組,來追蹤為以解決特定問題而訓練的所有模型。使用 AWS SDK for Python (Boto3) 或 Amazon SageMaker Studio 主控台建立模型群組。

建立模型群組 (Boto3)

重要

允許 Amazon SageMaker Studio 或 Amazon SageMaker Studio Classic 建立 Amazon SageMaker 資源的自訂IAM政策也必須授予許可,才能將標籤新增至這些資源。需要將標籤新增至資源的許可,因為 Studio 和 Studio Classic 會自動標記他們建立的任何資源。如果IAM政策允許 Studio 和 Studio Classic 建立資源,但不允許標記,則嘗試建立資源時可能會發生「AccessDenied」錯誤。如需詳細資訊,請參閱提供標記 SageMaker資源的許可

AWS Amazon 的受管政策 SageMaker 提供建立 SageMaker 資源的許可,已包含在建立這些資源時新增標籤的許可。

若要使用 Boto3 建立模型群組,請呼叫 create_model_package_groupAPI操作,並將名稱和描述指定為參數。下列範例展示如何建立模型群組。來自 create_model_package_group 呼叫的回應是新模型群組的 Amazon Resource Name (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']))

建立模型群組 (Studio 或 Studio Classic)

若要在 Amazon SageMaker Studio 主控台中建立模型群組,請根據您是否使用 Studio 或 Studio Classic 完成下列步驟。

Studio
  1. 請依照啟動 Amazon SageMaker Studio 中的指示開啟 Studio 主控台。 SageMaker

  2. 在左側導覽窗格中選擇 Models (模型)

  3. 如果尚未選取,請選擇已註冊模型索引標籤。

  4. 如果尚未選取,請在已註冊模型索引標籤標籤的正下方選擇模型群組

  5. 選擇註冊 ,然後選擇模型群組

  6. 註冊模型群組對話方塊中,輸入下列資訊:

    • 模型群組名稱欄位中新模型群組的名稱

    • (選用) 描述欄位中模型群組的描述

    • (選用) 您要與標籤欄位中的模型群組建立關聯的任何鍵值對。如需與使用標記相關的資訊,請參閱 AWS 一般參考中的標記 AWS 資源

  7. 選擇註冊模型群組

  8. (選用) 在模型頁面中,選擇已註冊模型索引標籤,然後選擇模型群組 。確認新建立的模型群組出現在模型群組清單中。

Studio Classic
  1. 登入 Amazon SageMaker Studio Classic。如需詳細資訊,請參閱啟動 Amazon SageMaker Studio Classic。

  2. 在左側的導覽窗格中,選擇首頁圖示 ( Black square icon representing a placeholder or empty image. )。

  3. 選擇模型,然後選擇模型註冊表

  4. 選擇動作,然後選擇 建立模型群組

  5. 建立模型群組對話方塊中,輸入下列資訊:

    • 模型群組名稱欄位中輸入新模型群組的名稱。

    • (可選) 在描述欄位中,輸入模型群組的描述。

    • (可選) 在標籤欄位中,輸入您希望與模型群組相關聯的任何鍵值對。如需與使用標記相關的資訊,請參閱 AWS 一般參考中的標記 AWS 資源

    • (可選) 在專案欄位中選擇要與模型群組相關聯的專案。如需與專案相關的資訊,請參閱MLOps 使用 SageMaker 專案自動化

  6. 選擇建立模型群組