├── .idea ├── .gitignore ├── PBXAI.iml ├── deployment.xml ├── dictionaries │ └── Danie.xml ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── README.md ├── resource ├── agent │ └── placeholder.txt ├── cache │ └── mimic │ │ └── placeholder.txt ├── knowledge_graph │ ├── kg.pkl │ ├── knowledge_graph_chinese.svg │ ├── knowledge_graph_english.svg │ ├── knowledge_graph_output.csv │ ├── knowledge_graph_script.txt │ └── knowledge_graph_script.txt.txt ├── mapping_file │ └── mimic │ │ ├── DIAGNOSIS.csv │ │ ├── DISTRIBUTION_CONVERT.csv │ │ ├── LABTEST_LIST.csv │ │ ├── MEDICINE_NAME_MAP.csv │ │ ├── OPERATION_MAP.csv │ │ └── feature_order.csv ├── preprocessed_data │ ├── mimic_feature_distribution │ │ └── placeholder.txt │ └── placeholder.txt ├── raw_data │ └── mimic │ │ └── placeholder.txt ├── representation │ └── placeholder.txt └── reward_set.csv └── src ├── data_preprocess ├── mimic_data_split.py ├── mimic_distribution_convert_and_impute.py ├── mimic_feature_selection.py ├── mimic_patient_feature_generator.py ├── mimic_visit_selection_and_reorganize.py └── util.py ├── experiment_util.py ├── model ├── kg_env.py ├── knowledge_graph.py ├── performance_eval.py └── train_agent.py └── representation_learning ├── concept_learning.py └── representation_learning.py /.idea/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/.idea/.gitignore -------------------------------------------------------------------------------- /.idea/PBXAI.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/.idea/PBXAI.iml -------------------------------------------------------------------------------- /.idea/deployment.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/.idea/deployment.xml -------------------------------------------------------------------------------- /.idea/dictionaries/Danie.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/.idea/dictionaries/Danie.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/README.md -------------------------------------------------------------------------------- /resource/agent/placeholder.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /resource/cache/mimic/placeholder.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /resource/knowledge_graph/kg.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/knowledge_graph/kg.pkl -------------------------------------------------------------------------------- /resource/knowledge_graph/knowledge_graph_chinese.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/knowledge_graph/knowledge_graph_chinese.svg -------------------------------------------------------------------------------- /resource/knowledge_graph/knowledge_graph_english.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/knowledge_graph/knowledge_graph_english.svg -------------------------------------------------------------------------------- /resource/knowledge_graph/knowledge_graph_output.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/knowledge_graph/knowledge_graph_output.csv -------------------------------------------------------------------------------- /resource/knowledge_graph/knowledge_graph_script.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/knowledge_graph/knowledge_graph_script.txt -------------------------------------------------------------------------------- /resource/knowledge_graph/knowledge_graph_script.txt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/knowledge_graph/knowledge_graph_script.txt.txt -------------------------------------------------------------------------------- /resource/mapping_file/mimic/DIAGNOSIS.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/mapping_file/mimic/DIAGNOSIS.csv -------------------------------------------------------------------------------- /resource/mapping_file/mimic/DISTRIBUTION_CONVERT.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/mapping_file/mimic/DISTRIBUTION_CONVERT.csv -------------------------------------------------------------------------------- /resource/mapping_file/mimic/LABTEST_LIST.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/mapping_file/mimic/LABTEST_LIST.csv -------------------------------------------------------------------------------- /resource/mapping_file/mimic/MEDICINE_NAME_MAP.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/mapping_file/mimic/MEDICINE_NAME_MAP.csv -------------------------------------------------------------------------------- /resource/mapping_file/mimic/OPERATION_MAP.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/mapping_file/mimic/OPERATION_MAP.csv -------------------------------------------------------------------------------- /resource/mapping_file/mimic/feature_order.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/mapping_file/mimic/feature_order.csv -------------------------------------------------------------------------------- /resource/preprocessed_data/mimic_feature_distribution/placeholder.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /resource/preprocessed_data/placeholder.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /resource/raw_data/mimic/placeholder.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /resource/representation/placeholder.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /resource/reward_set.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/resource/reward_set.csv -------------------------------------------------------------------------------- /src/data_preprocess/mimic_data_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/data_preprocess/mimic_data_split.py -------------------------------------------------------------------------------- /src/data_preprocess/mimic_distribution_convert_and_impute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/data_preprocess/mimic_distribution_convert_and_impute.py -------------------------------------------------------------------------------- /src/data_preprocess/mimic_feature_selection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/data_preprocess/mimic_feature_selection.py -------------------------------------------------------------------------------- /src/data_preprocess/mimic_patient_feature_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/data_preprocess/mimic_patient_feature_generator.py -------------------------------------------------------------------------------- /src/data_preprocess/mimic_visit_selection_and_reorganize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/data_preprocess/mimic_visit_selection_and_reorganize.py -------------------------------------------------------------------------------- /src/data_preprocess/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/data_preprocess/util.py -------------------------------------------------------------------------------- /src/experiment_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/experiment_util.py -------------------------------------------------------------------------------- /src/model/kg_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/model/kg_env.py -------------------------------------------------------------------------------- /src/model/knowledge_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/model/knowledge_graph.py -------------------------------------------------------------------------------- /src/model/performance_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/model/performance_eval.py -------------------------------------------------------------------------------- /src/model/train_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/model/train_agent.py -------------------------------------------------------------------------------- /src/representation_learning/concept_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/representation_learning/concept_learning.py -------------------------------------------------------------------------------- /src/representation_learning/representation_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZJU-BMI/PBXAI/HEAD/src/representation_learning/representation_learning.py --------------------------------------------------------------------------------