fix(Evaluation): fixed unequal length of training epochs by multi mod…

…el pipelines
frankkramer-lab · Jun 30, 2022 · 6729798 · 6729798
1 parent a3f0f3f
commit 6729798
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/aucmedi/evaluation/fitting.py b/aucmedi/evaluation/fitting.py
@@ -80,10 +80,11 @@ def evaluate_fitting(train_history, out_path, monitor=["loss"],
                                              train_history[key]
                     del train_history[key]
                     del train_history[sec_key]
-
-    # Convert to pandas dataframe
-    dt = pd.DataFrame.from_dict(train_history, orient="columns")
 
+    # Convert to pandas dataframe
+    hist_prepared = dict([ (k,pd.Series(v)) for k,v in train_history.items() ])
+    dt = pd.DataFrame.from_dict(hist_prepared, orient="columns")
+
     # Identify all selected columns
     selected_cols = []
     for key in train_history: