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

Question

使用 scikit learn 进行逻辑回归时，代码如下：

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 的含义。 感谢您的时间。

标签： pythonscikit-learnprediction

Answer 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)。