Python中的预测值

Question

有人能帮我得到预测概率值的代码吗？我的模型运行良好，给我的预测是1和0，但我也需要概率值。代码在两个python文件中。第一个文件使用训练数据集创建地图文件。第二个python文件（评分文件）使用测试数据上的map文件进行预测。有人能告诉我应该插入什么代码来得到概率值吗。 下面的代码来自于评分文件，这里我需要代码来获得概率值

pred = model.predict(X.values)
data["Predicted"] = pred

# I NEED THE CODE HERE TO GET THE PROBABILITY VALUES. 

data.to_excel(r'result.xlsx', index=False)

非常感谢

Answer 1

我无法再编辑我的问题，所以把它放在这里。在

谢谢你的帮助。我使用随机森林模型。这是我的代码，下面的第4行给出了一个错误。如果删除第4行，代码将运行，但在最终的excel文件中，我没有得到概率，只得到预测值1和0。有人能告诉我怎么解决这个错误吗。错误的最后一行说 ValueError:传递的项数错误2，放置意味着1

pred = model.predict(X.values)
data["Predicted"] = pred
prob = model.predict_proba(X.values)
data["Pred Value"]= prob   - this line causes error
data.to_excel(r'result.xlsx', index=False)' 

谢谢

Answer 2

检查您的模型是否有predict_proba方法。在

用法与相同的predict方法相同。在

prob = model.predict_proba(X.values)

编辑：

sklearn的一些学习模型实现提供了predict_proba方法。它不是一个指标，但正如我所说的，是一种班级学习模式的方法。在

例如：

^{pr2}$