Routine updates.

ProjectNeura · Sep 14, 2024 · d3ea103 · d3ea103
1 parent 1ed561b
commit d3ea103
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dataset/cxr_datamodule.py b/dataset/cxr_datamodule.py
@@ -29,7 +29,9 @@ def setup(self, stage):
             # split train/val
             msss = MultilabelStratifiedShuffleSplit(
                 n_splits=1, test_size=self.cfg["val_split"], random_state=self.cfg["seed"])
-            print(self.cfg["classes"], self.df[self.cfg["classes"]])
+            print(self.cfg["classes"])
+            print(self.df)
+            print(self.df[self.cfg["classes"]])
             train_idx, val_idx = next(msss.split(self.df, self.df[self.cfg["classes"]].values))
             train_df = self.df.iloc[train_idx]
             val_df = self.df.iloc[val_idx]