Fix label bug when labels share common string

src/caked/dataloader.py

@@ -119,7 +119,7 @@ def load(self, datapath, datatype) -> None:
             Path(datapath) / p
             for p in paths
             for c in self.classes
-            if c in p.split("_")[0]
+            if c == p.split("_")[0]
         ]
         if self.dataset_size is not None:
             paths = paths[: self.dataset_size]