예제 #1
0
 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
예제 #2
0
 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