model_config: Export Airflow model config from a SageMaker model

In DyfanJones/sagemaker-r-workflow: sagemaker pipeline and workflows

Export Airflow model config from a SageMaker model

Description

Export Airflow model config from a SageMaker model

Usage

model_config(model, instance_type = NULL, role = NULL, image_uri = NULL)

Arguments

model	(sagemaker.model.Model): The Model object from which to export the Airflow config
instance_type	(str): The EC2 instance type to deploy this Model to. For example, 'ml.p2.xlarge'
role	(str): The “ExecutionRoleArn“ IAM Role ARN for the model
image_uri	(str): An Docker image URI to use for deploying the model

Value

dict: Model config that can be directly used by SageMakerModelOperator in Airflow. It can also be part of the config used by SageMakerEndpointOperator and SageMakerTransformOperator in Airflow.

DyfanJones/sagemaker-r-workflow documentation built on April 3, 2022, 11:28 p.m.