From 6982e322a27e8b94873653d08da8791d67e07fff Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Fri, 26 Feb 2021 16:19:34 +0100
Subject: [PATCH] add annotation if loss is on val or test
---
mlair/run_modules/post_processing.py | 2 +-
mlair/run_modules/training.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/mlair/run_modules/post_processing.py b/mlair/run_modules/post_processing.py
index 85b272cf..807f32bb 100644
--- a/mlair/run_modules/post_processing.py
+++ b/mlair/run_modules/post_processing.py
@@ -391,7 +391,7 @@ class PostProcessing(RunEnvironment):
path = self.data_store.get("model_path")
with open(os.path.join(path, "test_scores.txt"), "a") as f:
for index, item in enumerate(test_score):
- logging.info(f"{self.model.metrics_names[index]}, {item}")
+ logging.info(f"{self.model.metrics_names[index]} (test), {item}")
f.write(f"{self.model.metrics_names[index]}, {item}\n")
def train_ols_model(self):
diff --git a/mlair/run_modules/training.py b/mlair/run_modules/training.py
index 4409b643..bbb3fabf 100644
--- a/mlair/run_modules/training.py
+++ b/mlair/run_modules/training.py
@@ -250,4 +250,4 @@ class Training(RunEnvironment):
val_score = self.model.evaluate_generator(generator=self.val_set, use_multiprocessing=True, verbose=0, steps=1)
for index, item in enumerate(val_score):
- logging.info(f"{self.model.metrics_names[index]}, {item}")
+ logging.info(f"{self.model.metrics_names[index]} (val), {item}")
--
GitLab