备忘：区分tensorflow中几种cross_entropy函数

最新推荐文章于 2021-05-21 22:13:29 发布

观生生不息

最新推荐文章于 2021-05-21 22:13:29 发布

阅读量326

点赞数

分类专栏：备忘 tensorflow

本文链接：https://blog.csdn.net/RambleMY/article/details/104731317

版权

备忘同时被 2 个专栏收录

13 篇文章 0 订阅

订阅专栏

tensorflow

6 篇文章 0 订阅

订阅专栏

# A logit (also called a score) is a raw unscaled value associated with a class.
# 多分类问题 输入标签需要Onehot编码 输出logits向量
tf.nn.softmax_cross_entropy_with_logits
# 多分类问题 输入标签不必Onehot编码 输出logits向量
tf.nn.sparse_softmax_cross_entropy_with_logits

# 多分类问题 输入标签需要Onehot编码 输出loss值
tf.losses.softmax_cross_entropy
# 多分类问题 输入标签不必Onehot编码 输出loss值
tf.losses.sparse_softmax_cross_entropy

# DEPRECATED
tf.contrib.losses.softmax_cross_entropy
tf.conbrib.losses.sparse_softmax_cross_entropy

# 关系
tf.losses.softmax_cross_entropy = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits)
# 有权重
tf.losses.softmax_cross_entropy(weight=W) = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits*W)

# The sigmoid is a partial case of the softmax function
# 二分类
tf.nn.sigmoid_cross_entropy_with_logits
tf.nn.weighted_cross_entropy_with_logits(labels, logits, pos_weight, name=None)
tf.losses.sigmoid_cross_entropy
tf.contrib.losses.sigmoid_cross_entropy(DEPRECATED)