如何通过Sagemaker管道步骤向模型注册表中添加自定义元数据？

Question

【以下的问题经过翻译处理】 基于Sagemaker Python SDK，我可以通过以下示例代码添加自定义/客户元数据。我计划使用Sagemaker模型步骤并将模型注册到模型包组中创建端到端管道。在模型创建或注册模型时，是否有可能添加这些元数据？

Answer

【以下的回答经过翻译处理】 是的，这是可以的。您可以在SageMaker SDK的Model对象的[register](https://sagemaker.readthedocs.io/en/stable/api/inference/model.html#sagemaker.model.Model.register)方法中添加自定义元数据。然后，可以使用SageMaker Pipelines的[Model阶段](https://docs.aws.amazon.com/sagemaker/latest/dg/build-and-manage-steps.html#step-type-model)与此注册方法一起使用。

以下是从[此页面](https://docs.aws.amazon.com/sagemaker/latest/dg/build-and-manage-steps.html#step-type-model)提供的示例中复制的代码片段，其中我添加了具有您的示例输入的`custom_metadata_properties`参数。

```
register_model_step_args = pipeline_model.register(
    content_types=["application/json"],
    response_types=["application/json"],
    inference_instances=["ml.t2.medium", "ml.m5.xlarge"],
    transform_instances=["ml.m5.xlarge"],
    model_package_group_name="sipgroup",
    customer_metadata_properties={"key1": "value1", "key2": "value2"}
)

step_model_registration = ModelStep(
    name="AbaloneRegisterModel",
    step_args=register_model_step_args,
)
```

如何通过Sagemaker管道步骤向模型注册表中添加自定义元数据？

Contenuto pertinente