minor bugs

notebooks @ abdb0696

+Subproject commit abdb06965783302b8d5dd5d7f8bdb8e3cc9d639c

src/scripts/extract_metrics.py

@@ -28,6 +28,7 @@ def extract_summaries_to_csv(root_dir,metrics=None,force=False):
                 print(f"Skipping {root} as metrics.csv already exists. Use --force to overwrite")
                 continue
             #log_dir = "../../../data-project/KIWaVo/models/lfu/willenscharen6/lightning_logs/version_0/"
+            try :
                 reader_hp = SummaryReader(root,pivot=True)
                 df =reader_hp.scalars
                 df = df.drop(["step","epoch"],axis=1)[1:49]
@@ -35,6 +36,8 @@ def extract_summaries_to_csv(root_dir,metrics=None,force=False):
                 df.columns = df.columns.str.slice(3)
                 df.to_csv(root + "/metrics.csv")
                 print(f"Extracted metrics for {root}")
+            except Exception as e:
+                print(f"Error in {root}")
 
 
 def main():

src/utils/helpers.py

@@ -88,7 +88,7 @@ def get_pred(
     y_pred = np.concatenate(pred)
 
     if y_true is not None:
-        y_pred = np.concatenate([np.expand_dims(y_true, axis=1), pred], axis=1)
+        y_pred = np.concatenate([np.expand_dims(y_true, axis=1), y_pred], axis=1) # Replaced pred with y_pred, why was it pred?
         y_pred = pd.DataFrame(y_pred, index=y_true.index, columns=range(y_pred.shape[1]))
     return y_pred