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

由bqthdmgv创建，最终由small_q更新于2024-11-18 02:14 被浏览 14 用户

我尝试用系统提供的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, 线下可以预测概率。