我正在测试SageMaker AutoPilot，以验证其日常使用效果如何。
到目前为止，使用它似乎相对容易，它训练了具有良好结果的模型，并且很容易创建端点。我想获得预测的标签及其概率，以检查谓词是否良好。但是，我只能得到标签，而检索概率（predict_proba）却一无所获。
有什么方法可以得到这个概率吗？谢谢！
答案 0 :(得分：2)
您现在可以选择分类模型的推理响应：）
答案 1 :(得分：0)
在创建端点模型时，必须在容器列表中的每个容器中设置环境变量SAGEMAKER_INFERENCE_OUTPUT和SAGEMAKER_INFERENCE_INPUT。有效值为：'predicted_label, probabilit(y)(ies),labels'
请参阅以下内容以了解容器配置：https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.create_model
朱利安（Julien）发布的链接也有一些不错的信息，但信息也不完整：https://docs.aws.amazon.com/sagemaker/latest/dg/autopilot-automate-model-development-container-output.html