diff --git a/code/bolsonaro/trainer.py b/code/bolsonaro/trainer.py index e615ca8c97facd1b8e033e60eec11c1cc8de922e..a9bebe044b68475f5cc0cf6c6a2097ffe986e47c 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 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391