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