model.predict_proba()和model.predict()冲突行为

Question

我有两班一班和二班：

model.predict_proba()的输出：

[0.333,0.6667]

model.predict()的输出：

1

在10个lac的测试数据中，大约200个测试值正在发生这种情况。对于所有其他记录，这两个函数的输出并不冲突。即使2类的概率较高，预测函数也给出最后一类为1。

对于这200个记录，model.predict()和model.predict_proba()的输出不同步的原因是什么？

Answer 1

如果您有两个类，即您正在进行二进制分类，则应该将目标标签指定为0或1 (neg/pos)。predict_proba()方法返回样本在第1类中的估计概率。

“在二进制分类情况下，'1‘表示阳性样本，'0’表示负样本，我们还支持0,1中的概率值作为标签，以表示实例为正的概率。”