Merge branch 'f/fix_split_kvasir' into 'main'

Fixed correct train val test split, by getting validation set as proportion... See merge request es/ai/hannah/hannah!386
ekut-es · May 14, 2024 · cae1a1f · cae1a1f
2 parents b557b34 + 0ed02e7
commit cae1a1f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/hannah/datasets/vision/kvasir.py b/hannah/datasets/vision/kvasir.py
@@ -192,8 +192,9 @@ def process_official_split(df: pd.DataFrame):
             train_images = split0_paths
             train_labels = split0_labels
 
-            val_images = split1_paths
-            val_labels = split1_labels
+            train_images, val_images, train_labels, val_labels = train_test_split(
+                train_images, train_labels, test_size=0.25, random_state=1
+            )
 
             test_images = split1_paths
             test_labels = split1_labels