了解`predict_proba`的结果

Posted 2023-03-12

【中文标题】了解`predict_proba`的结果

【英文标题】：Understanding results of `predict_proba`

【发布时间】：2020-05-07 10:56:50

【问题描述】：

我使用深度神经网络进行预测（可以是 1 或 0）。解决的问题是分类问题。我正在使用sklearn 的predict_proba 来预测我的预测概率。据我所知，predict_proba 的输出每行应有 2 列，表示每个类的概率。 我期待的是这样的：

[0.99467199 0.00532801]
[0.00223842 0.99776158]
[0.99389073 0.00610927]

但我得到了：

[[6.92488432e-01]
 [5.56170940e-04]
 [7.24569559e-01]
 [9.99984384e-01]]

如何解释这些结果？这些对应于类别概率的值是否被预测？

【问题讨论】：

在你的问题中添加相关代码sn-p

@AkshayNevrekar 这是一个研究项目，我不能在这里给出模型代码。我还要在这里添加什么以使问题更清晰？

【参考方案1】：

这是正常的，当你只有两个类时，只给出其中一个的概率就足够了，因为可以通过从 1 中减去第一个类的概率来找到另一个。

【讨论】：

那么我的输出显示了什么？它是显示 0（或 1）类的概率还是最终预测的类的概率？