--- a
+++ b/app/datasets/base.py
@@ -0,0 +1,11 @@
+import pickle
+
+
+def load_data(dataset_type):
+    # Load data
+    data_path = f"datasets/{dataset_type}/processed_data/"
+    x = pickle.load(open(data_path + "x.pkl", "rb"))
+    y = pickle.load(open(data_path + "y.pkl", "rb"))
+    x_lab_length = pickle.load(open(data_path + "visits_length.pkl", "rb"))
+
+    return x, y, x_lab_length