首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >将tf.layers转换为tf.keras.layers后的成本函数

将tf.layers转换为tf.keras.layers后的成本函数
EN

Stack Overflow用户
提问于 2019-03-11 22:56:12
回答 2查看 38关注 0票数 0

我有一个CNN,它的输出维度是None,10

这是一个多标签问题,其中输出表示x可能属于的可能类别。(例如,可以将图像分类为cat dark等)

以下是我现在所拥有的,如何将代码更改为keras版本?我找不到等同于sigmoid_cross_entropy_with_logits

代码语言:javascript
复制
 model = tf.layers.dense(L3, category_num, activation=None)
 cross_entropy = tf.nn.sigmoid_cross_entropy_with_logits(logits=model, labels=Y)

 cost = tf.reduce_mean(tf.reduce_sum(cross_entropy, axis=1))
 optimizer = tf.train.AdamOptimizer(learning_rate).minimize(cost)
EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2019-03-12 00:06:42

Keras中的直接替代方法是在输出层使用sigmoid激活,并将binary_crossentropy用作成本函数。

代码语言:javascript
复制
net.add(Dense(..., activation='sigmoid'))
net.compile(optimizer, loss='binary_crossentropy')

看一看https://github.com/keras-team/keras/issues/741

票数 0
EN

Stack Overflow用户

发布于 2019-03-12 00:45:57

在Keras中:

代码语言:javascript
复制
#you model here -- last layer:

model.add(Dense(10))
model.add(Activation('sigmoid'))
model.compile(loss='categorical_crossentropy',
              optimizer="adam",metrics=['accuracy'])
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/55104664

复制
相关文章

相似问题

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