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

0

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

profile picture
EXPERTO
preguntada hace un año71 visualizaciones
1 Respuesta
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
EXPERTO
respondido hace un año

No has iniciado sesión. Iniciar sesión para publicar una respuesta.

Una buena respuesta responde claramente a la pregunta, proporciona comentarios constructivos y fomenta el crecimiento profesional en la persona que hace la pregunta.

Pautas para responder preguntas