Keras eval callback

callback.py

import keras


class EvalCallback(keras.callbacks.Callback):

    def __init__(self, X_val, y_val, metric):
        super().__init__()
        self.X_val = X_val
        self.y_val = y_val
        self.metric = metric

    def on_epoch_end(self, epoch, logs={}):
        y_pred = self.model.predict(self.X_val, verbose=0)
        score = self.metric(self.y_val, y_pred)
        print('\nEpoch {} score: {:.6f}\n'.format(epoch+1, score))