【问题标题】:Understanding results of `predict_proba`了解`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 这是一个研究项目,我不能在这里给出模型代码。我还要在这里添加什么以使问题更清晰?

标签: scikit-learn deep-learning classification prediction tensorflow-probability


【解决方案1】:

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

【讨论】:

  • 那么我的输出显示了什么?它是显示 0(或 1)类的概率还是最终预测的类的概率?
猜你喜欢
  • 2022-01-22
  • 2019-03-20
  • 2012-01-25
  • 1970-01-01
  • 2011-04-10
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 2021-04-24
相关资源
最近更新 更多