모델 패키지 단계 및 상태 업데이트 예시(boto3)

모델 패키지 단계 및 상태를 업데이트하려면 관련 권한이 있는 실행 역할을 수임해야 합니다. 다음은 AWS SDK for Python (Boto3)을 사용하여 UpdateModelPackage API를 통해 단계 상태를 업데이트하는 방법에 대한 예시입니다.

이 예시에서 UpdateModelPackage API 작업에 대한 "Development"라는 ModelLifeCycle 단계 및 "Approved" 단계 상태 조건 키가 실행 역할에 부여되었습니다. stage-description에 설명을 포함할 수 있습니다. 자세한 정보는 스테이징 구조 설정 예시을 참조하세요.

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)

model_package_update_input_dict = {
    "ModelLifeCycle" : { 
        "stage" : "Development",
        "stageStatus" : "Approved",
        "stageDescription" : "stage-description"
    }
} 
model_package_update_response = sm_client.update_model_package(**model_package_update_input_dict)