From c1e168fde120ea23bec37bbee5bd4ab536d1649b Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Tue, 11 Oct 2022 11:38:15 +0200
Subject: [PATCH] debug eval script

---
 mu_map/eval/measures.py | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/mu_map/eval/measures.py b/mu_map/eval/measures.py
index 28661f7..426cef1 100644
--- a/mu_map/eval/measures.py
+++ b/mu_map/eval/measures.py
@@ -72,7 +72,7 @@ if __name__ == "__main__":
     args = parser.parse_args()
 
     if args.split == "all":
-        args.split == None
+        args.split = None
 
     torch.set_grad_enabled(False)
 
@@ -83,7 +83,7 @@ if __name__ == "__main__":
 
     transform_normalization = SequenceTransform(
         transforms=[
-            norm_by_str(args.normalization),
+            norm_by_str(args.norm),
             PadCropTranform(dim=3, size=args.size),
         ]
     )
@@ -109,17 +109,9 @@ if __name__ == "__main__":
     print(f" " * 100, end="\r")
 
     values = dict(map(lambda x: (x[0], np.array(x[1])), values.items()))
-    scores_mse = np.array(scores_mse)
-    scores_nmae = np.array(scores_nmae)
-
-    mse_avg = scores_mse.mean()
-    mse_std = np.std(scores_mse)
-
-    nmae_avg = scores_nmae.mean()
-    nmae_std = np.std(scores_nmae)
 
     print("Scores:")
     for measure_name, measure_values in values.items():
         mean = measure_values.mean()
         std = np.std(measure_values)
-        print(f" - {measure_name}: {mean:.6f}±{std:.6f}")
+        print(f" - {measure_name:>6}: {mean:.6f}±{std:.6f}")
-- 
GitLab