问答交流

【平台使用】系统提供的XGBClassifier的predict_proba的预测结果只有0和1

由bqthdmgv创建,最终由small_q 被浏览 12 用户

我尝试用系统提供的XGBClassifier的predict_proba()来做预测,以便后面根据预测结果排序,结果发现系统提供的该功能只能预测0或1. 线下在自己的电脑上,该功能可以预测出0到1之间的任何值。

yp=model.predict_proba(x)
yp

array([[1., 0.],
       [1., 0.],
       [1., 0.],
       ...,
       [1., 0.],
       [1., 0.],
       [1., 0.]], dtype=float32)

\

评论
  • 检查一下标签设置和模型参数,如果是二分类问题可能预测就是0和1,如果还有问题可以把代码分享看一下
  • 按照XGBoost文档,无论是二分类还是多分类,只要用predict_proba()函数,应该返回概率,用predict()函数,就返回0或1,线下也确实验证了这一点。
  • 代码就是上面两行,不需要多余代码。模型是线下训练好上传到平台的。平台预测的结果只有0,1, 线下可以预测概率。
{link}