diff --git a/mlair/run_modules/post_processing.py b/mlair/run_modules/post_processing.py index 85b272cf170322a78b69c594030d8cd96101f61e..807f32bb828d9a37b8fa94c9e6bdedd7dfeabf90 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 4409b6438ccaf6a46dc21aaf359425ea52095279..bbb3fabf4dd7c62a71db04cc982544b61d097240 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}")