首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >Keras softmax概率

Keras softmax概率
EN

Stack Overflow用户
提问于 2017-09-18 06:22:48
回答 1查看 672关注 0票数 0

我试图从最后一层是softmax层的网络中获得softmax概率,当我使用model.predict()时,我得到的是类而不是概率。谁能告诉我如何获得概率。

代码语言:javascript
复制
 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'))
EN

回答 1

Stack Overflow用户

发布于 2018-06-19 22:22:58

你的模型的输出将是0到1之间的值。你的模型应该给出一个大小为43的向量,所有输出的总和将加到1。

根据您的训练,如果与训练示例相似,这些“概率”对于所选的类通常几乎是1,这表明模型经过了良好的训练。

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/46269423

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档