diff --git a/Experiments/8-Uncertainty_graph.py b/Experiments/8-Uncertainty_graph.py
index 4635eca6fe8440473c84ed6404b440494e4b49b4..f53ff10d93d476b2d6fc1c92606ff5a82aa04aab 100644
--- a/Experiments/8-Uncertainty_graph.py
+++ b/Experiments/8-Uncertainty_graph.py
@@ -21,7 +21,7 @@ clean_df = clean_table(excel_dir=excel_dir, pre_mRS=2)
 
 # Given a clean table get features and labels
 table = TableReader(input_df=clean_df, tables=['all_timepoints'], data_dictionaries='timepoints', mv_strategy='median',
-                    output_feature=['mortality'])
+                    output_feature=['dmRS'])
 
 output_vector = table.output_vector
 
@@ -77,7 +77,7 @@ for k in range(5):
                                    one_hot=True)
     ages_fold = dataloader_fold.meta['Age'].values[fold_indices[k][2]]
 
-    """config_graph = {
+    config_graph = {
         'Age': True,
         'beta_Age': 3,
         'Sex': False,
@@ -86,8 +86,9 @@ for k in range(5):
         'NIHSS': False,
         'beta_NIHSS': 3
 
-    }"""
-    config_graph = {
+    }
+
+    """config_graph = {
         'Age': True,
         'beta_Age': 4,
         'Sex': True,
@@ -97,7 +98,10 @@ for k in range(5):
         'beta_NIHSS': 3
 
     }
-    fold_name = "C:/Users/martinca1/PhD/Projects/AI_Stroke/out/models/EdgeDropout3_mortality"
+    """
+    #fold_name = "C:/Users/martinca1/PhD/Projects/AI_Stroke/out/models/EdgeDropout3_mortality"
+
+    fold_name = "C:/Users/martinca1/PhD/Projects/AI_Stroke/out/models/EdgeDropout03"
     loader = dataloader_fold.build_graph(config_graph=config_graph)
     torch.manual_seed(0)
     for i in range(10):
diff --git a/__pycache__/evaluate_model.cpython-37.pyc b/__pycache__/evaluate_model.cpython-37.pyc
index 322b92ce507f58b1850b31118dd901c41cf6da33..c40b25887db2f091f0095d5bb596ea68ba706a5a 100644
Binary files a/__pycache__/evaluate_model.cpython-37.pyc and b/__pycache__/evaluate_model.cpython-37.pyc differ