我如何知道特定类的输出?

时间:2019-06-11 12:20:21

标签: conv-neural-network multiclass-classification

我训练了一个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,但我不确定

0 个答案:

没有答案
相关问题