首页 > 其他 > 详细

Tensorflow 之 loss

时间:2020-02-25 09:39:40      阅读:63      评论:0      收藏:0      [点我收藏+]

参考博客:https://blog.csdn.net/ZJRN1027/article/details/80199248

1、计算cost的过程

  1)对神经网络的输出(logits)进行softmax,即概率归一化;

  技术分享图片

 

 

  2)结合样本的标签labels计算交叉熵cross-entropy,作为loss;

  技术分享图片

  3)对batch中所有样本的loss进行平均得到cost

 

2、tensorflow中计算loss的函数

  1)tf.nn.sparse.softmax_cross_entropy_with_logits():输入的label格式为1维的向量

  2)tf.nn.softmax_cross_entropy_with_logits():输入的label格式为one-hot格式,即矩阵

 

 

计算Cross-Entropy

Tensorflow 之 loss

原文:https://www.cnblogs.com/wt-seu/p/12360079.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!