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

你可能感兴趣的:(TensorFlow)