Skip to content

Instantly share code, notes, and snippets.

@andreaschandra
Last active May 14, 2018 08:31
Show Gist options
  • Save andreaschandra/ed59f0929ba10cdd9063010722a98460 to your computer and use it in GitHub Desktop.
Save andreaschandra/ed59f0929ba10cdd9063010722a98460 to your computer and use it in GitHub Desktop.
kmeans_model = 'topic-kmeans'
sm = boto3.client('sagemaker')
containers = {'us-west-2': '174872318107.dkr.ecr.us-west-2.amazonaws.com/kmeans:latest'}
create_model_response = sm.create_model( ModelName=kmeans_model,
ExecutionRoleArn=role,
PrimaryContainer={
'Image': containers[boto3.Session().region_name],
'ModelDataUrl': 's3://{}/{}/model.tar.gz'.format(bucket, prefix)
}
)
print(create_model_response['ModelArn'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment