--- a
+++ b/sybil/utils/helpers.py
@@ -0,0 +1,18 @@
+from sybil.datasets.validation import CSVDataset
+from sybil.datasets.nlst import *
+from sybil.datasets.mgh import MGH_Screening
+
+SUPPORTED_DATASETS = {
+    "validation": CSVDataset,
+    "nlst": NLST_Survival_Dataset,
+    "nlst_risk_factors": NLST_Risk_Factor_Task,
+    "nlst_for_plco2012": NLST_for_PLCO,
+    "nlst_for_plco2019": NLST_for_PLCO_Screening,
+    "mgh": MGH_Screening,
+}
+
+
+def get_dataset(dataset_name, split, args):
+    if dataset_name not in SUPPORTED_DATASETS:
+        raise NotImplementedError("Dataset {} does not exist.".format(dataset_name))
+    return SUPPORTED_DATASETS[dataset_name](args, split)