我是靠谱客的博主 心灵美纸飞机,这篇文章主要介绍tf.keras.losses.SparseCategoricalCrossentropy,现在分享给大家,希望可以做个参考。

交叉熵损失函数

tf.keras.losses.SparseCategoricalCrossentropy(
    from_logits=False, reduction=losses_utils.ReductionV2.AUTO,
    name='sparse_categorical_crossentropy'
)

y_pred为N维向量(N为类别的个数),label为单个数字,如果label也是one-hot之后的值,需要使用CategoricalCrossentropy损失函数。

参数:

  • from_logits: 为True时,会将y_pred转化为概率(用softmax),否则不进行转换,通常情况下用True结果更稳定;
  • reduction:类型为tf.keras.losses.Reduction,对loss进行处理,默认是求平均;
  • name: op的name

最后

以上就是心灵美纸飞机最近收集整理的关于tf.keras.losses.SparseCategoricalCrossentropy的全部内容,更多相关tf内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(39)

评论列表共有 0 条评论

立即
投稿
返回
顶部