我训练了一个CNN模型,该模型是一个多分类器。我在输出层中使用了softmax单位,但是现在,我不知道与各个概率相关的类。
我正在使用model.predict_proba来获取概率矩阵
class_mode ='类别', classes = [“ zipper”,“ backstrap”,“ slip_on”,“ lace_up”])
如果model.predict_proba的输出= [0.2,0.2,0.2,0.4] 哪个类别对应于0.4?
我认为它遵循字母顺序。 backstrap = 0,lace_up = 1,slip_on = 2,拉链= 3,但我不确定