交叉熵损失函数 (tf、keras)
tf.keras.losses.SparseCategoricalCrossentropy(
from_logits=False,
reduction=losses_utils.ReductionV2.AUTO,
name='sparse_categorical_crossentropy'
)
参数:
- from_logits: 为True时,会将y_pred转化为概率(用softmax),否则不进行转换,通常情况下用True结果更稳定;
- reduction:类型为tf.keras.losses.Reduction,对loss进行处理,默认是求平均;
- name: op的name
PS:
y_pred为N维向量(N为类别的个数),label为单个数字,如果label也是one-hot之后的值,需要使用CategoricalCrossentropy损失函数。