Custom Loss function Keras Tensorflow

Viewed 9927

I need a custom weighted MSE loss function. I defined it in keras.backend

from keras import backend as K
def weighted_loss(y_true, y_pred):
    return K.mean( K.square(y_pred - y_true) *
    K.exp(-K.log(1.7) * (K.log(1. + K.exp((y_true - 3)/5 ))))      
    ,axis=-1  )

However, a test run returns

    weighted_loss(1,2)
ValueError: Tensor conversion requested dtype int32 for Tensor with dtype float32: 'Tensor("Exp_37:0", shape=(), dtype=float32)'  

or

    weighted_loss(1.,2.)
ZeroDivisionError: integer division or modulo by zero   

I wonder what mistakes am I making here.

1 Answers
Related