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