Keras softmax概率

时间:2017-09-17 22:22:48

标签: keras softmax

我试图从最后一层是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'))

1 个答案:

答案 0 :(得分:0)

您的模型的输出将是介于0和1之间的值。 您的模型应给出一个大小为43的矢量,所有输出的总和将加一。

根据您的训练,如果与训练示例类似,则所选班级的这些“概率”通常几乎是一个,这表明模型已经训练有素。