交叉熵损失函数
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内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复