tf.keras.losses.sparse_categorical_crossentropy

介绍

多分类交叉熵函数,标签是1,2,3,非one-hot值

计算方式类似:tf.keras.losses.categorical_crossentropy

常用别名

  • tf.losses.sparse_categorical_crossentropy

字符串表示:"sparse_categorical_crossentropy"

语法

tf.keras.losses.sparse_categorical_crossentropy(
    y_true,
    y_pred,
    from_logits=False,
    axis=-1
)
参数
  • y_true:真实值
  • y_pred:预测值
  • from_logits=False:默认值为False,表示y_pred是经过softmax函数计算的结果。如果 y_pred 没有经过softmax函数计算,需要设置为True
  • axis=-1:整数,指定通道轴。axis=-1对应于数据格式channels_last’
返回值

tensor


原文出处:/show_1EF5F1ygQuiB.html