Sagemaker存储模型评估度量

Question

目前，我正在建立作为我们的团队ML平台的Sagemaker，我们希望将模型评估度量与模型本身一起存储。

现在，我们正在使用boto3 using客户端：https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.create_model创建模型。

但是我不知道如何将度量附加到它上，所以我们可以运行模型评估并存储度量来跟踪它们。

我们的目标是在Sagemaker中建立模型，这样我们就可以将其部署为端点。

提前感谢！

Answer 1

如果使用create_model_package API在模型注册表中注册度量，则可以将度量附加到模型。API中有一个名为ModelMetrics的参数，您可以使用它。度量必须存储在S3中的一个文件中，您将在这个API调用中提供S3 URI。

https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateModelPackage.html