Added Poisson loss to Objectives

2015-07-20 16:09:53 -07:00 · 2015-07-20 16:09:53 -07:00 · 149d0e8d18
commit 149d0e8d18
parent 37965cae6b
1 changed files with 3 additions and 0 deletions
--- a/keras/objectives.py
+++ b/keras/objectives.py
@ -49,6 +49,9 @@ def binary_crossentropy(y_true, y_pred):
    bce = T.nnet.binary_crossentropy(y_pred, y_true).mean(axis=-1)
    return bce

+def poisson_loss(y_true, y_pred):
+    return T.mean(y_pred - y_true * T.log(y_pred), axis=-1)
+
 # aliases
 mse = MSE = mean_squared_error
 mae = MAE = mean_absolute_error