Skip to content

Instantly share code, notes, and snippets.

@gleeb
Created August 7, 2022 17:46
Show Gist options
  • Save gleeb/fb9dd086434eb0e7c1a97d5865238545 to your computer and use it in GitHub Desktop.
Save gleeb/fb9dd086434eb0e7c1a97d5865238545 to your computer and use it in GitHub Desktop.
Sagemaker reate package group
import os
import sagemaker
import boto3
iam_client = boto3.client('iam')
role = iam_client.get_role(RoleName='AmazonSageMaker-ExecutionRole-xyz')['Role']['Arn']
sm_client = boto3.client('sagemaker', region_name="us-east-1")
model_package_group_name = "Titanic-survival"
model_package_group_input_dict = {
"ModelPackageGroupName" : model_package_group_name,
"ModelPackageGroupDescription" : "Sample model package for predicting titanic survival rate"
}
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']))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment