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


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


计算 y_true 和 y_pred 之间的泊松损失。

用法

tf.keras.metrics.poisson(
    y_true, y_pred
)

参数

  • y_true 基本事实值。形状 = [batch_size, d0, .. dN]
  • y_pred 预测值。形状 = [batch_size, d0, .. dN]

返回

  • 泊松损失值。形状 = [batch_size, d0, .. dN-1]

抛出

  • InvalidArgumentError 如果 y_truey_pred 的形状不兼容。

泊松损失是 Tensor y_pred - y_true * log(y_pred) 的元素的平均值。

单机使用:

y_true = np.random.randint(0, 2, size=(2, 3))
y_pred = np.random.random(size=(2, 3))
loss = tf.keras.losses.poisson(y_true, y_pred)
assert loss.shape == (2,)
y_pred = y_pred + 1e-7
assert np.allclose(
    loss.numpy(), np.mean(y_pred - y_true * np.log(y_pred), axis=-1),
    atol=1e-5)

相关用法


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