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

0

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

profile picture
专家
已提问 8 个月前57 查看次数
1 回答
0

【以下的回答经过翻译处理】 是的,这是可以的。您可以在SageMaker SDK的Model对象的register方法中添加自定义元数据。然后,可以使用SageMaker Pipelines的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,
)
profile picture
专家
已回答 8 个月前

您未登录。 登录 发布回答。

一个好的回答可以清楚地解答问题和提供建设性反馈,并能促进提问者的职业发展。

回答问题的准则