Debug

ThomasMeissnerDS · Aug 3, 2024 · bfde201 · bfde201
1 parent e12b444
commit bfde201
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bluecast/general_utils/general_utils.py b/bluecast/general_utils/general_utils.py
@@ -206,7 +206,7 @@ def save_out_of_fold_data(
         elif isinstance(y_classes, list):
             y_classes = np.asarray(y_classes).astype(int)
 
-        y_true = y_true.astype(int)
+        y_true = y_true.astype(int).flatten()
 
         print("DEBUUUUUUUUUUUUUUUUG MULTICLASS")
         print(y_true.shape)