当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.keras.metrics.sparse_categorical_crossentropy用法及代码示例


计算稀疏分类交叉熵损失。

用法

tf.keras.metrics.sparse_categorical_crossentropy(
    y_true, y_pred, from_logits=False, axis=-1
)

参数

  • y_true 基本事实值。
  • y_pred 预测值。
  • from_logits y_pred 是否预期为 logits 张量。默认情况下，我们假设 y_pred 对概率分布进行编码。
  • axis 默认为 -1。计算熵的维度。

返回

  • 稀疏分类交叉熵损失值。

单机使用：

y_true = [1, 2]
y_pred = [[0.05, 0.95, 0], [0.1, 0.8, 0.1]]
loss = tf.keras.losses.sparse_categorical_crossentropy(y_true, y_pred)
assert loss.shape == (2,)
loss.numpy()
array([0.0513, 2.303], dtype=float32)

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.keras.metrics.sparse_categorical_crossentropy。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。