我试图从最后一层是softmax图层的网络中获取softmax概率,当我使用model.predict()时,我得到了类而不是概率。任何人都可以告诉我们如何获得概率。
model = Sequential()
model.add(Convolution2D(32, 3, 3,input_shape=(32, 32, 3)))
model.add(MaxPooling2D((2, 2)))
model.add(Dropout(0.5))
model.add(Activation('relu'))
model.add(Convolution2D(32, 3, 3))
model.add(MaxPooling2D((2, 2)))
model.add(Dropout(0.5))
model.add(Activation('relu'))
model.add(Flatten())
model.add(Dense(128))
model.add(Activation('relu'))
model.add(Dense(43))
model.add(Activation('softmax'))
答案 0 :(得分:0)
您的模型的输出将是介于0和1之间的值。 您的模型应给出一个大小为43的矢量,所有输出的总和将加一。
根据您的训练,如果与训练示例类似,则所选班级的这些“概率”通常几乎是一个,这表明模型已经训练有素。