From 1271fc5e469d91aad6fc23d0a0a0a145beb10d61 Mon Sep 17 00:00:00 2001 From: Chao Pang Date: Fri, 29 Nov 2024 17:20:48 -0500 Subject: [PATCH] Missing output names (#76) * fixed the streaming bug * rewrite the key from test to validation to be consistent with dataset names * added the missing output_names.py --- src/cehrbert/config/output_names.py | 9 +++++++++ 1 file changed, 9 insertions(+) create mode 100644 src/cehrbert/config/output_names.py diff --git a/src/cehrbert/config/output_names.py b/src/cehrbert/config/output_names.py new file mode 100644 index 0000000..53e3dab --- /dev/null +++ b/src/cehrbert/config/output_names.py @@ -0,0 +1,9 @@ +parquet_data_path = "patient_sequence" +qualified_concept_list_path = "qualified_concept_list" +time_attention_model_path = "time_aware_model.h5" +bert_model_validation_path = "bert_model.h5" +mortality_data_path = "mortality" +heart_failure_data_path = "heart_failure" +hospitalization_data_path = "hospitalization" +information_content_data_path = "information_content" +concept_similarity_path = "concept_similarity"