30-01-22

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):