final commit - 4real4real

Author: SagiPolaczek

demo.py

@@ -1,5 +1,17 @@
 from epsilon_runner import *
 
+"""
+
+----------------------------
+If you wish to 'play' with the demo/runner, please change the following values in the 'epsilon_runner.py' source file.
+For changes in the hyper parameters consider changing: TRAIN_COMMON_PARAMS, INFER_COMMON_PARAMS, EVAL_COMMON_PARAMS.
+
+For further modifications please check "fuse-med-ml" documentation and source code at: https://github.com/BiomedSciAI/fuse-med-ml,
+or contact us.
+
+----------------------------
+
+
 
 experiment = "full"  # Choose from supported experiments
 
@@ -10,10 +22,9 @@
     "overlap",
 ]
 
-assert experiment in supported_experiments, f"runner doesn't support experiment ({experiment})."
-
+# Set you paths
 
-ROOT = "./_demo"
+ROOT = "./_examples/epsilon" 
 model_dir = os.path.join(ROOT, f"model_dir_{experiment}")
 PATHS = {
     "model_dir": model_dir,
@@ -24,6 +35,8 @@
     "data_split_filename": os.path.join(ROOT, "eps_split.pkl"),
 }
 
+"""
+
 
 if __name__ == "__main__":
 

epsilon_runner.py

@@ -94,11 +94,11 @@
 
 
 model_dir = os.path.join(ROOT, f"model_dir_{experiment}")
-cache_suffix = "_MLP"
+
 PATHS = {
     "model_dir": model_dir,
-    "cache_dir_train": os.path.join(ROOT, f"cache_dir_train{cache_suffix}"),
-    "cache_dir_eval": os.path.join(ROOT, f"cache_dir_eval{cache_suffix}"),
+    "cache_dir_train": os.path.join(ROOT, f"cache_dir_train"),
+    "cache_dir_eval": os.path.join(ROOT, f"cache_dir_eval"),
     "inference_dir": os.path.join(model_dir, "infer"),
     "eval_dir": os.path.join(model_dir, "eval"),
     "data_split_filename": os.path.join(ROOT, "eps_split.pkl"),