我正在测试SageMaker AutoPilot，以验证它是否适合常规使用。
到目前为止，使用它似乎相对容易，它训练了一个模型并取得了良好的结果，并且很容易创建端点。我想获得预测的标签及其概率，以检查预测是否良好。但是，我只能获得标签，我没有找到任何关于检索概率(predict_proba(的信息。
有什么方法可以得到概率吗？谢谢！
您现在可以选择分类模型的推理响应:)
更多信息请访问 https://docs.aws.amazon.com/sagemaker/latest/dg/autopilot-automate-model-development-container-output.html
创建终端节点模型时，必须在容器列表的每个容器中设置环境变量：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
朱利安发布的链接也有一些很好但不完整的信息：https://docs.aws.amazon.com/sagemaker/latest/dg/autopilot-automate-model-development-container-output.html