From 7ef1daad0699e52df7db7baaa1509bff8b5a59f0 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Thu, 6 Jun 2024 11:19:27 +0200
Subject: [PATCH] [classification.experiment] Save trainlog in train directory
---
src/mednet/libs/classification/scripts/experiment.py | 5 +++--
.../libs/classification/tests/test_cli_classification.py | 2 +-
2 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/src/mednet/libs/classification/scripts/experiment.py b/src/mednet/libs/classification/scripts/experiment.py
index f15f4c7d..3a469a8c 100644
--- a/src/mednet/libs/classification/scripts/experiment.py
+++ b/src/mednet/libs/classification/scripts/experiment.py
@@ -99,10 +99,11 @@ def experiment(
logger.info("Started train analysis")
from mednet.libs.common.scripts.train_analysis import train_analysis
+ logdir = train_output_folder / "logs"
ctx.invoke(
train_analysis,
- logdir=train_output_folder / "logs",
- output_folder=output_folder / "trainlog.pdf",
+ logdir=logdir,
+ output_folder=train_output_folder,
)
logger.info("Ended train analysis")
diff --git a/src/mednet/libs/classification/tests/test_cli_classification.py b/src/mednet/libs/classification/tests/test_cli_classification.py
index 8eb8a93c..7e81d045 100644
--- a/src/mednet/libs/classification/tests/test_cli_classification.py
+++ b/src/mednet/libs/classification/tests/test_cli_classification.py
@@ -456,7 +456,7 @@ def test_experiment(temporary_basedir):
)
== 1
)
- assert (output_folder / "trainlog.pdf").exists()
+ assert (output_folder / "model" / "trainlog.pdf").exists()
assert (
len(
list(
--
GitLab