initial work for elpv and subter datasets

elpv as example dataset/implementation
subter with final dataset

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

@@ -1,4 +1,6 @@
 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 .odds import ODDSADDataset
@@ -19,6 +21,8 @@ def load_dataset(
 
     implemented_datasets = (
         "mnist",
+        "elpv",
+        "subter",
         "fmnist",
         "cifar10",
         "arrhythmia",
@@ -32,6 +36,22 @@ def load_dataset(
 
     dataset = None
 
+    if dataset_name == "subter":
+        dataset = SubTer_Dataset(
+            root=data_path,
+            ratio_known_normal=ratio_known_normal,
+            ratio_known_outlier=ratio_known_outlier,
+            ratio_pollution=ratio_pollution,
+        )
+
+    if dataset_name == "elpv":
+        dataset = ELPV_Dataset(
+            root=data_path,
+            ratio_known_normal=ratio_known_normal,
+            ratio_known_outlier=ratio_known_outlier,
+            ratio_pollution=ratio_pollution,
+        )
+
     if dataset_name == "mnist":
         dataset = MNIST_Dataset(
             root=data_path,