fix(Evaluation): added support for only transfer learning epochs in e…

…valuate_fitting()
frankkramer-lab · Jul 18, 2022 · e474282 · e474282
1 parent 9e75baf
commit e474282
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/aucmedi/evaluation/fitting.py b/aucmedi/evaluation/fitting.py
@@ -111,7 +111,8 @@ def evaluate_fitting(train_history,
             tl_epochs = filter_tl.groupby(["prefix"])["epoch"].max()
             # compute fine tune epoch update
             group_repeats = filter.groupby(["prefix"]).size()
-            ft_update = tl_epochs.repeat(group_repeats).to_numpy()
+            if group_repeats.empty : ft_update = 0
+            else : ft_update = tl_epochs.repeat(group_repeats).to_numpy()
         # if no prefix available -> add epochs to all ft phases
         else:
             # identify number of epochs global