首页 > 解决方案 > 在 AWS SageMaker AutoPilot 端点上预测概率

问题描述

我正在测试 SageMaker AutoPilot,以验证它对于常规使用的效果。

到目前为止,它似乎相对容易使用,它训练了一个效果很好的模型,并且很容易创建端点。我想获得预测的标签及其概率,以检查预测是否良好。但是,我只能得到标签,我没有找到任何关于检索概率(predict_proba)的信息。

有什么方法可以得到概率吗?谢谢!

标签: python-3.xamazon-web-servicesamazon-sagemaker

解决方案


您现在可以选择分类模型的推理响应 :)

更多信息,请访问https://docs.aws.amazon.com/sagemaker/latest/dg/autopilot-automate-model-development-container-output.html


推荐阅读