在用Xgboost时,可以使用xgb.predict_proba()输出各个类型的预测概率,但出现预测的各个类型概率一致的情况,你有可能是犯了以下错误:
模型初始化没有注意加predictor参数,该参数默认和你使用GPU还是CPU训练模型有关,若你使用GPU训练模型,改参数默认为’gpu_predictor’,假如你用cpu训练模型,默认就为’cpu_predictor’。当你使用GPU跑出模型之后,用于CPU预测,就会出现概率一致情况,需加上该参数重新训练模型。详情可以参考xgboost官方文档 https://xgboost.readthedocs.io/en/latest/gpu/index.html(附:官方文档并没有指出会出现什么样的错误…)
Xgboost预测predict_proba()输出概率一致

相关文章:

  • 2021-08-27
  • 2019-08-30
  • 2021-11-20
  • 2021-06-26
  • 2021-09-28
  • 2021-07-12
  • 2021-12-16
  • 2021-08-18
猜你喜欢
  • 2022-12-23
  • 2022-01-22
  • 2021-12-19
  • 2021-10-08
  • 2022-12-23
  • 2021-05-14
  • 2021-07-13
相关资源
相似解决方案