def on_epoch_end(self, epoch, logs={}): prtr = ev.eval_sts(self.model.predict(self.train_gr)['classes'], loader.sts_categorical2labels(self.train_gr['classes']), 'Train', quiet=True).Pearson prval = ev.eval_sts(self.model.predict(self.val_gr)['classes'], loader.sts_categorical2labels(self.val_gr['classes']), 'Val', quiet=True).Pearson print(' train Pearson %f val Pearson %f' % (prtr, prval)) logs['pearson'] = prval