XGB:展示每个预测结果的各特征概率

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了XGB:展示每个预测结果的各特征概率相关的知识,希望对你有一定的参考价值。

参考技术A 问题描述:

我正在尝试使用xgboost算法预测解决多类分类,但是我不知道 predict_proba 是如何工作的.实际上, predict_proba 会生成一个概率列表,但我不知道每种概率与哪个类别相关.

这是一个简单的例子:

这是我的训练数据:

然后当我尝试预测probas的新例子

这将返回如下内容:

我的问题是:概率为0.5的类别是什么?是2类,3类还是0类?

似乎您在使用xgboost的sklearn API.在这种情况下,模型具有专用的属性 model.classes _ ,该属性返回模型学习到的类,并且输出数组中类的顺序与概率的顺序相对应.

这是一个有关伪数据的示例:

以上是关于XGB:展示每个预测结果的各特征概率的主要内容,如果未能解决你的问题,请参考以下文章

如何获得逻辑回归特征对特定预测的相对重要性?

xgboost predict 方法为所有行返回相同的预测值

xgboost predict 方法为所有行返回相同的预测值

如何在文本分类中使用朴素贝叶斯预测所需的类

朴素贝叶斯分类和预测算法的原理及实现

逻辑回归算法的原理及实现(LR)