【发布时间】: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 这是一个研究项目,我不能在这里给出模型代码。我还要在这里添加什么以使问题更清晰?
标签: scikit-learn deep-learning classification prediction tensorflow-probability