如何通过lambda创建和执行sagemaker管道?

Question

基于文档，我想创建一个带有注册模型步骤(下面的示例)的sagemaker管道，并通过lambda执行它。有例子吗?我需要下载现有的model.tar.gz文件，重新打包注册。

from sagemaker.model import Model
from sagemaker.inputs import CreateModelInput
from sagemaker.workflow.steps import CreateModelStep
from sagemaker.model_metrics import MetricsSource, ModelMetrics
from sagemaker.workflow.step_collections import RegisterModel
model = Model(
image_uri=image_uri, # the XGBoost image
model_data=step_train.properties.ModelArtifacts.S3ModelArtifacts, # The S3 location of the 'model.tar.gz' file
sagemaker_session=sagemaker_session,
role=role,
)
inputs = CreateModelInput(
instance_type="ml.m5.large",
accelerator_type="ml.eia1.medium",
)
step_create_model = CreateModelStep(
name="adultCreateModel",
model=model,
inputs=inputs,
)
step_register = RegisterModel(
name="adultRegisterModel",
estimator=xgb_train,
model_data=step_train.properties.ModelArtifacts.S3ModelArtifacts, # The S3 uri to the 'model.tar.gz' file
content_types=["text/csv"],
response_types=["text/csv"],
inference_instances=["ml.t2.medium", "ml.m5.xlarge"],
transform_instances=["ml.m5.xlarge"],
)

Answer 1

有关使用Lambda创建SageMaker Pipeline的指导，请参阅本博客- https://aws.amazon.com/blogs/machine-learning/use-a-sagemaker-pipeline-lambda-step-for-lightweight-model-deployments/