Breaking News: Grepper is joining You.com. Read the official announcement!

how to use rmse as loss function in keras

Darkstar answered on September 12, 2021 Popularity 7/10 Helpfulness 10/10

answer how to use rmse as loss function in keras

how to use rmse as loss function in keras

Comment

from keras import backend as K

def root_mean_squared_error(y_true, y_pred):
        return K.sqrt(K.mean(K.square(y_pred - y_true), axis=-1)) 

  model.compile(optimizer = "rmsprop", loss = root_mean_squared_error, 
              metrics =["accuracy"])

Popularity 7/10 Helpfulness 10/10 Language python

Source: stackoverflow.com

Tags: keras loss-function python

Link to this answer
Share Copy Link