Add model column to evaluation result table

joeynmt_server/models/evaluation_results.py

@@ -7,6 +7,7 @@ class EvaluationResult(BaseModel):
     __tablename__ = 'evaluation_results'
 
     label = db.Column(db.Unicode(50), nullable=False)
+    model = db.Column(db.Unicode(500), nullable=False)
     correct = db.Column(db.Integer, nullable=False)
     total = db.Column(db.Integer, nullable=False)
 

joeynmt_server/trainer.py

@@ -175,8 +175,10 @@ def train_n_rounds(config_basename, min_rounds=10):
                 correct = round(accuracy * total)
                 logging.info('Got validation result: {}/{} = {}.'
                              .format(correct, total, accuracy))
-                evr = EvaluationResult(label='running_dev', correct=correct,
-                                       total=total)
+                evr = EvaluationResult(
+                    label='running_dev', model=config_basename,
+                    correct=correct, total=total
+                )
                 db.session.add(evr)
                 db.session.commit()
                 logging.info('Saving parses from running dev set.')
@@ -242,8 +244,8 @@ def validate(config_basename, dataset_name='dev'):
         correct = round(accuracy * total)
         logging.info('Got validation result: {}/{} = {}.'
                      .format(correct, total, accuracy))
-        evr = EvaluationResult(label=dataset_name, correct=correct,
-                               total=total)
+        evr = EvaluationResult(label=dataset_name, model=config_basename,
+                               correct=correct, total=total)
         db.session.add(evr)
         db.session.commit()
     except: