LightGBM多分类预测结果

问题描述 投票:0回答:1

我尝试使用lightGBM建立多分类模型。训练模型后,我在线分析了一些数据并将其放入模型中进行预测。

但是,结果对我来说似乎很奇怪。我认为我的预测的嵌套数组意味着每个类的概率(我有4个类)。 x-test(我用于验证的数据)的结果似乎正确。但是我抓取的数据结果似乎很奇怪。总计不等于1。

prediction result

在此帖子multiclass-classification-with-lightgbm中,预测结果也未加1!

这两个数据框在我看来是相同的,并且我使用的是完全相同的模型!有人可以告诉我如何解释结果或我做错了什么吗?x-test dataframedata I scraped dataframe

python machine-learning predict multiclass-classification lightgbm
1个回答
0
投票

对于多类别分类,when the classes are not mutually exclusive,概率之和可能不等于1。假设您要对图像中的狗,猫和鸟进行分类,但模型显示为汽车图像,则这三个类别的概率应较低且不等于1。如果需要,可以使用this formula重新调整预测的比例强制概率之和为1。

例如,当您具有类型为1 vs others的分类器时,图像只能是猫,狗或鸟。在这种情况下,这些类是互斥的,并且概率之和应为1。

更多参考文献ref1ref2ref3

© www.soinside.com 2019 - 2024. All rights reserved.