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

0

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

profile picture
EXPERTE
gefragt vor einem Jahr71 Aufrufe
1 Antwort
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
EXPERTE
beantwortet vor einem Jahr

Du bist nicht angemeldet. Anmelden um eine Antwort zu veröffentlichen.

Eine gute Antwort beantwortet die Frage klar, gibt konstruktives Feedback und fördert die berufliche Weiterentwicklung des Fragenstellers.

Richtlinien für die Beantwortung von Fragen