Scikit 学习输出 predict_proba 和 predict. 的含义

【问题标题】：Scikit Learn the meaning of output **predict_proba** and **predict.**Scikit 学习输出 **predict_proba** 和 **predict.** 的含义
【发布时间】：2020-01-21 01:46:04
【问题描述】：

使用scikit learn进行Logistic Regression时，代码如下：

print(sentiment_model.predict_proba(test_features[:2]))
print(sentiment_model.predict(test_features[:2]))

输出：

[[1.20150248e-04 9.99879850e-01]
 [9.99986356e-01 1.36436948e-05]]
[1 0]

我很难理解输出 predict_proba 和 predict. 的含义。感谢您的时间。

【问题讨论】：

标签： python scikit-learn prediction

【解决方案1】：

在 Scikit-Learn 中，predict 返回预测（0 或 1、False 或 True），而 predict_proba 返回预测的确定性分数。 predict_proba 的输出是一个数组，第一个元素是实例是 0 类（或 False 类）的概率估计，第二个元素是实例是概率的估计1 类（或 True 类）。因此，predict_proba 输出中的行加起来为 1。

在您的代码中，第一个实例是 1 类（无论在您的应用程序中编码的含义是什么），概率为 0.999879859。然而，在第二种情况下，它是1 类的概率很小（0.00001364...）。因此，它是一个0 类，概率为0.99998635)。

【讨论】：

感谢您的澄清