介绍
多分类交叉熵函数,标签是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