predicted_proba在多类物流分类中



我有 5 个类作为输出,代表学生的预测成绩 - 1、2、3、4、5。

当我输入我的值时,用predict_proba,我得到每个类的概率。

例如。

1级:0.1级 2级:0.4级:3

:0.3级 4级:0.05级 5级:0.15

但是,我从中得到的唯一结果是预测是:类 2,因为概率最高。如何将这些概率转换为更具体的预测,例如 2.4 或 2.9?其中 2 等于类 2

这可能吗?如果是这样,如何?

非常感谢您的时间。

这是一种快速的方法:

probs = LRC.predict_proba(x)
class_indexes = np.argmax(probs,axis=1)
max_probs = probs[np.arange(len(x)),class_indexes]
class_output = lrc.classes_[class_indexes]
class_prob_list = zip(class_output,max_probs)
print [str(cls)+'.'+str(prob) for cls,prob in class_prob_list]

相关内容

  • 没有找到相关文章