Only call 'softmax_cross_entropy_with_logits' with named grguments(labels=...,logits=...)

来源:互联网 发布:好搜小说软件 编辑:程序博客网 时间:2024/06/03 19:13

运行一段代码:

with tf.name_scope('cross_entropy'):        cross_entropy = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(y,y_))        tf.summary.scalar('cross_entropy',cross_entropy)



将其改为:cross_entropy = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=y, labels=y_))

补充交叉熵知识:

        交叉熵可在神经网络(机器学习)中作为损失函数,p表示真实标记的分布,q则为训练后的模型的预测标记分布,交叉熵损失函数可以衡量p与q的相似性。交叉熵作为损失函数还有一个好处是使用sigmoid函数在梯度下降时能避免均方误差损失函数学习速率降低的问题,因为学习速率可以被输出的误差所控制。tensorflow中自带的函数可以轻松的实现交叉熵的计算。

tf.nn.softmax_cross_entropy_with_logits(_sentinel=None, labels=None, logits=None, dim=-1, name=None)
注意:如果labels的每一行是one-hot表示,也就是只有一个地方为1,其他地方为0,可以使用tf.sparse_softmax_cross_entropy_with_logits()
警告:
1. 这个操作的输入logits是未经缩放的,该操作内部会对logits使用softmax操作
2. 参数labels,logits必须有相同的形状 [batch_size, num_classes] 和相同的类型(float16, float32, float64)中的一种
参数:_sentinel: 一般不使用
labels: labels实际的标签,大小同上的每一行labels[i]必须为一个概率分布
logits: 就是神经网络最后一层的输出,如果有batch的话,它的大小就是[batchsize,num_classes],单样本的话,大小就是num_classes
dims: 类的维度,默认-1,也就是最后一维
name: 该操作的名称
返回值:长度为batch_size的一维Tensor



阅读全文
1 0
原创粉丝点击