Python tf.keras.losses.SparseCategoricalCrossentropy用法及代码示例

计算标签和预测之间的交叉熵损失。

继承自：Loss

用法

tf.keras.losses.SparseCategoricalCrossentropy(
    from_logits=False, reduction=losses_utils.ReductionV2.AUTO,
    name='sparse_categorical_crossentropy'
)

参数

from_logits y_pred 是否预期为 logits 张量。默认情况下，我们假设 y_pred 对概率分布进行编码。
reduction 类型tf.keras.losses.Reduction适用于损失。默认值为AUTO.AUTO表示缩减选项将由使用上下文确定。对于几乎所有情况，这默认为SUM_OVER_BATCH_SIZE.当与tf.distribute.Strategy，在内置训练循环之外，例如tf.keras compile和fit，使用AUTO或者SUM_OVER_BATCH_SIZE将引发错误。请参阅此自定义训练教程更多细节。
name 实例的可选名称。默认为“sparse_categorical_crossentropy”。

当有两个或多个标签类时使用此交叉熵损失函数。我们希望标签以整数形式提供。如果您想使用one-hot 表示提供标签，请使用CategoricalCrossentropy loss。对于 y_pred ，每个特征应该有 # classes 浮点值，对于 y_true ，每个特征应该有一个浮点值。

在下面的代码段中，每个示例都有一个浮点值 y_true 和 # classes 每个示例的浮点值 y_pred 。 y_true 的形状是 [batch_size] ， y_pred 的形状是 [batch_size, num_classes] 。

单机使用：

y_true = [1, 2]
y_pred = [[0.05, 0.95, 0], [0.1, 0.8, 0.1]]
# Using 'auto'/'sum_over_batch_size' reduction type.
scce = tf.keras.losses.SparseCategoricalCrossentropy()
scce(y_true, y_pred).numpy()
1.177

# Calling with 'sample_weight'.
scce(y_true, y_pred, sample_weight=tf.constant([0.3, 0.7])).numpy()
0.814

# Using 'sum' reduction type.
scce = tf.keras.losses.SparseCategoricalCrossentropy(
    reduction=tf.keras.losses.Reduction.SUM)
scce(y_true, y_pred).numpy()
2.354

# Using 'none' reduction type.
scce = tf.keras.losses.SparseCategoricalCrossentropy(
    reduction=tf.keras.losses.Reduction.NONE)
scce(y_true, y_pred).numpy()
array([0.0513, 2.303], dtype=float32)

compile() API 的用法：

model.compile(optimizer='sgd',
              loss=tf.keras.losses.SparseCategoricalCrossentropy())

相关用法

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