SageMakerRegisterModelVersionOperator
AmazonRegisters an Amazon SageMaker model by creating a model version that specifies the model group to which it belongs. Will create the model group if it does not exist already.
Access Instructions
Install the Amazon provider package into your Airflow environment.
Import the module into your DAG file and instantiate it with your desired params.
Parameters
image_uriRequiredThe Amazon EC2 Container Registry (Amazon ECR) path where inference code is stored.
model_urlRequiredThe Amazon S3 path where the model artifacts (the trained weights of the model), which result from model training, are stored. This path must point to a single gzip compressed tar archive (.tar.gz suffix).
package_group_nameRequiredThe name of the model package group that the model is going to be registered to. Will be created if it doesn’t already exist.
package_group_descDescription of the model package group, if it was to be created (optional).
package_descDescription of the model package (optional).
model_approvalApproval status of the model package. Defaults to PendingManualApproval
extrasCan contain extra parameters for the boto call to create_model_package, and/or overrides for any parameter defined above. For a complete list of available parameters, see https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.create_model_package
str
Documentation
Registers an Amazon SageMaker model by creating a model version that specifies the model group to which it belongs. Will create the model group if it does not exist already.
See also
For more information on how to use this operator, take a look at the guide: Register a Sagemaker Model Version