fix(evaluation): fix evaluate training charts

frankkramer-lab · Jul 15, 2022 · 87e3f9d · 87e3f9d
1 parent e6348e5
commit 87e3f9d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/aucmedi/evaluation/fitting.py b/aucmedi/evaluation/fitting.py
@@ -115,7 +115,9 @@ def evaluate_fitting(train_history,
         # if no prefix available -> add epochs to all ft phases
         else:
             # identify number of epochs global
-            tl_epochs = dt_melted["epoch"].max()
+            filter_tl = dt_melted[dt_melted["metric"].str.startswith("tl_")]
+            tl_epochs = filter_tl["epoch"].max()
+
             # compute fine tune epoch update
             ft_update = tl_epochs