tf.keras.losses.Poisson 损失函数 示例

泊松损失函数

预测值为为 $P$ ,真实值 $T$ 。

$poisson(P,T)=\frac{1}{n}{\sum }_{i=1}^n(P_i-T_{i}log(P_i))$

import tensorflow as tf

y_true = [[0., 1.], [0., 0.]]
y_pred = [[1., 1.], [0., 0.]]
# Using 'auto'/'sum_over_batch_size' reduction type.
p = tf.keras.losses.Poisson()
p(y_true, y_pred).numpy()

0.49999997