From f7e772d966f9de9ff100728ca1cd53cb106ec2a1 Mon Sep 17 00:00:00 2001
From: Luc Giffon <luc.giffon@lis-lab.fr>
Date: Thu, 21 Nov 2019 19:35:41 +0100
Subject: [PATCH] add experiments directory and print score on train and dev
---
code/bolsonaro/trainer.py | 6 ++++++
experiments/.gitkeep | 0
2 files changed, 6 insertions(+)
create mode 100644 experiments/.gitkeep
diff --git a/code/bolsonaro/trainer.py b/code/bolsonaro/trainer.py
index e615ca8..a9bebe0 100644
--- a/code/bolsonaro/trainer.py
+++ b/code/bolsonaro/trainer.py
@@ -74,3 +74,9 @@ class Trainer(object):
results.save(models_dir)
self._logger.info("Base performance on test: {}".format(results.test_score_regressor))
self._logger.info("Performance on test: {}".format(results.test_score))
+
+ self._logger.info("Base performance on train: {}".format(results.train_score_regressor))
+ self._logger.info("Performance on train: {}".format(results.train_score))
+
+ self._logger.info("Base performance on dev: {}".format(results.dev_score_regressor))
+ self._logger.info("Performance on dev: {}".format(results.dev_score))
diff --git a/experiments/.gitkeep b/experiments/.gitkeep
new file mode 100644
index 0000000..e69de29
--
GitLab