我有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]

10-08 02:07