From fd0a940844359592e1c9dc611c83b4c1315308b3 Mon Sep 17 00:00:00 2001
From: Simon Will <will@cl.uni-heidelberg.de>
Date: Sun, 21 Mar 2021 23:03:59 +0100
Subject: [PATCH] Fix bug in evaluation after training
---
joeynmt_server/trainer.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/joeynmt_server/trainer.py b/joeynmt_server/trainer.py
index d0252ae..47f45e9 100644
--- a/joeynmt_server/trainer.py
+++ b/joeynmt_server/trainer.py
@@ -168,10 +168,10 @@ def train_n_rounds(config_basename, min_rounds=10):
if dev:
dev_set = make_dataset_from_feedback(dev, model)
logging.info('Validating on {} feedback pieces.'
- .format(len(dev_set)))
+ .format(len(dev_set)))
results = model.validate(dev_set)
- correct = results['score']
total = len(dev_set)
+ correct = results['score'] * total
logging.info('Got validation result: {}/{} = {}'
.format(correct, total, results['score']))
evr = EvaluationResult(label='changing_dev', correct=correct,
--
GitLab