split subter implementation (training + inference)

Deep-SAD-PyTorch/src/datasets/main.py

@@ -1,9 +1,10 @@
-from .mnist import MNIST_Dataset
-from .elpv import ELPV_Dataset
-from .subter import SubTer_Dataset
-from .fmnist import FashionMNIST_Dataset
 from .cifar10 import CIFAR10_Dataset
+from .elpv import ELPV_Dataset
+from .fmnist import FashionMNIST_Dataset
+from .mnist import MNIST_Dataset
 from .odds import ODDSADDataset
+from .subter import SubTer_Dataset
+from .subtersplit import SubTerSplit_Dataset
 
 
 def load_dataset(
@@ -24,6 +25,7 @@ def load_dataset(
         "mnist",
         "elpv",
         "subter",
+        "subtersplit",
         "fmnist",
         "cifar10",
         "arrhythmia",
@@ -46,6 +48,15 @@ def load_dataset(
             inference=inference,
         )
 
+    if dataset_name == "subtersplit":
+        dataset = SubTerSplit_Dataset(
+            root=data_path,
+            ratio_known_normal=ratio_known_normal,
+            ratio_known_outlier=ratio_known_outlier,
+            ratio_pollution=ratio_pollution,
+            inference=inference,
+        )
+
     if dataset_name == "elpv":
         dataset = ELPV_Dataset(
             root=data_path,