有人能帮我得到预测概率值的代码吗?我的模型运行良好,给我的预测是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)
非常感谢
我无法再编辑我的问题,所以把它放在这里。在
谢谢你的帮助。我使用随机森林模型。这是我的代码,下面的第4行给出了一个错误。如果删除第4行,代码将运行,但在最终的excel文件中,我没有得到概率,只得到预测值1和0。有人能告诉我怎么解决这个错误吗。错误的最后一行说 ValueError:传递的项数错误2,放置意味着1
谢谢
检查您的模型是否有
predict_proba
方法。在用法与相同的
predict
方法相同。在编辑:
sklearn的一些学习模型实现提供了
predict_proba
方法。它不是一个指标,但正如我所说的,是一种班级学习模式的方法。在例如:
^{pr2}$相关问题 更多 >
编程相关推荐