├── .gitignore ├── AICare-baselines ├── .gitignore ├── README.md ├── configs │ ├── config.yaml │ ├── exp.py │ ├── experiments_challenge.py │ └── experiments_mimic.py ├── datasets │ ├── loader │ │ ├── datamodule.py │ │ ├── load_los_info.py │ │ └── unpad.py │ ├── mimic-iv │ ├── preprocess │ │ ├── statistics.py │ │ └── tools.py │ └── tjh ├── dl_tune.py ├── dl_tune2.py ├── losses │ ├── __init__.py │ ├── multitask_loss.py │ └── time_aware_loss.py ├── metrics │ ├── __init__.py │ ├── binary_classification_metrics.py │ ├── es.py │ ├── osmae.py │ ├── regression_metrics.py │ └── utils.py ├── models │ ├── __init__.py │ ├── adacare.py │ ├── agent.py │ ├── aicare.py │ ├── bi_lstm.py │ ├── catboost.py │ ├── concare.py │ ├── dt.py │ ├── gbdt.py │ ├── grasp.py │ ├── gru.py │ ├── gru_attn.py │ ├── heads.py │ ├── lightgbm.py │ ├── lr.py │ ├── lstm.py │ ├── mcgru.py │ ├── mlp.py │ ├── mt_rhn.py │ ├── retain.py │ ├── rf.py │ ├── rnn.py │ ├── stagenet.py │ ├── tcn.py │ ├── transformer.py │ ├── utils.py │ └── xgboost.py ├── pipelines │ ├── __init__.py │ ├── dl_pipeline.py │ └── ml_pipeline.py ├── requirements.txt ├── test.py └── train.py ├── README.md ├── config └── config_example.py ├── datasets ├── mimic-iv │ ├── extract_nshot.ipynb │ └── preprocess_mimic4.ipynb ├── preprocess │ ├── statistics.py │ └── tools.py └── tjh │ ├── extract_nshot.ipynb │ ├── preprocess_tjh.ipynb │ └── promptgen_tjh.ipynb ├── metrics ├── __init__.py ├── binary_classification_metrics.py ├── metrics_utils.py └── regression_metrics.py ├── performance.py ├── prompts ├── mimic-iv │ ├── 1-1_example.txt │ ├── N-1_batches_example.txt │ ├── N-1_list_example.txt │ ├── range.json │ ├── string_example.txt │ ├── type.json │ └── unit.json ├── prompt.py ├── template.txt └── tjh │ ├── 1-1_example.txt │ ├── N-1_batches_example.txt │ ├── N-1_list_example.txt │ ├── range.json │ ├── string_example.txt │ ├── type.json │ └── unit.json └── run.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/.gitignore -------------------------------------------------------------------------------- /AICare-baselines/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/.gitignore -------------------------------------------------------------------------------- /AICare-baselines/README.md: -------------------------------------------------------------------------------- 1 | # AICare-baselines -------------------------------------------------------------------------------- /AICare-baselines/configs/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/configs/config.yaml -------------------------------------------------------------------------------- /AICare-baselines/configs/exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/configs/exp.py -------------------------------------------------------------------------------- /AICare-baselines/configs/experiments_challenge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/configs/experiments_challenge.py -------------------------------------------------------------------------------- /AICare-baselines/configs/experiments_mimic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/configs/experiments_mimic.py -------------------------------------------------------------------------------- /AICare-baselines/datasets/loader/datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/datasets/loader/datamodule.py -------------------------------------------------------------------------------- /AICare-baselines/datasets/loader/load_los_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/datasets/loader/load_los_info.py -------------------------------------------------------------------------------- /AICare-baselines/datasets/loader/unpad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/datasets/loader/unpad.py -------------------------------------------------------------------------------- /AICare-baselines/datasets/mimic-iv: -------------------------------------------------------------------------------- 1 | /data/wangzx/llm4healthcare/datasets/mimic-iv -------------------------------------------------------------------------------- /AICare-baselines/datasets/preprocess/statistics.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /AICare-baselines/datasets/preprocess/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/datasets/preprocess/tools.py -------------------------------------------------------------------------------- /AICare-baselines/datasets/tjh: -------------------------------------------------------------------------------- 1 | /data/wangzx/llm4healthcare/datasets/tjh -------------------------------------------------------------------------------- /AICare-baselines/dl_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/dl_tune.py -------------------------------------------------------------------------------- /AICare-baselines/dl_tune2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/dl_tune2.py -------------------------------------------------------------------------------- /AICare-baselines/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/losses/__init__.py -------------------------------------------------------------------------------- /AICare-baselines/losses/multitask_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/losses/multitask_loss.py -------------------------------------------------------------------------------- /AICare-baselines/losses/time_aware_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/losses/time_aware_loss.py -------------------------------------------------------------------------------- /AICare-baselines/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/metrics/__init__.py -------------------------------------------------------------------------------- /AICare-baselines/metrics/binary_classification_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/metrics/binary_classification_metrics.py -------------------------------------------------------------------------------- /AICare-baselines/metrics/es.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/metrics/es.py -------------------------------------------------------------------------------- /AICare-baselines/metrics/osmae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/metrics/osmae.py -------------------------------------------------------------------------------- /AICare-baselines/metrics/regression_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/metrics/regression_metrics.py -------------------------------------------------------------------------------- /AICare-baselines/metrics/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/metrics/utils.py -------------------------------------------------------------------------------- /AICare-baselines/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/__init__.py -------------------------------------------------------------------------------- /AICare-baselines/models/adacare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/adacare.py -------------------------------------------------------------------------------- /AICare-baselines/models/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/agent.py -------------------------------------------------------------------------------- /AICare-baselines/models/aicare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/aicare.py -------------------------------------------------------------------------------- /AICare-baselines/models/bi_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/bi_lstm.py -------------------------------------------------------------------------------- /AICare-baselines/models/catboost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/catboost.py -------------------------------------------------------------------------------- /AICare-baselines/models/concare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/concare.py -------------------------------------------------------------------------------- /AICare-baselines/models/dt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/dt.py -------------------------------------------------------------------------------- /AICare-baselines/models/gbdt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/gbdt.py -------------------------------------------------------------------------------- /AICare-baselines/models/grasp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/grasp.py -------------------------------------------------------------------------------- /AICare-baselines/models/gru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/gru.py -------------------------------------------------------------------------------- /AICare-baselines/models/gru_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/gru_attn.py -------------------------------------------------------------------------------- /AICare-baselines/models/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/heads.py -------------------------------------------------------------------------------- /AICare-baselines/models/lightgbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/lightgbm.py -------------------------------------------------------------------------------- /AICare-baselines/models/lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/lr.py -------------------------------------------------------------------------------- /AICare-baselines/models/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/lstm.py -------------------------------------------------------------------------------- /AICare-baselines/models/mcgru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/mcgru.py -------------------------------------------------------------------------------- /AICare-baselines/models/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/mlp.py -------------------------------------------------------------------------------- /AICare-baselines/models/mt_rhn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/mt_rhn.py -------------------------------------------------------------------------------- /AICare-baselines/models/retain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/retain.py -------------------------------------------------------------------------------- /AICare-baselines/models/rf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/rf.py -------------------------------------------------------------------------------- /AICare-baselines/models/rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/rnn.py -------------------------------------------------------------------------------- /AICare-baselines/models/stagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/stagenet.py -------------------------------------------------------------------------------- /AICare-baselines/models/tcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/tcn.py -------------------------------------------------------------------------------- /AICare-baselines/models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/transformer.py -------------------------------------------------------------------------------- /AICare-baselines/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/utils.py -------------------------------------------------------------------------------- /AICare-baselines/models/xgboost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/models/xgboost.py -------------------------------------------------------------------------------- /AICare-baselines/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/pipelines/__init__.py -------------------------------------------------------------------------------- /AICare-baselines/pipelines/dl_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/pipelines/dl_pipeline.py -------------------------------------------------------------------------------- /AICare-baselines/pipelines/ml_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/pipelines/ml_pipeline.py -------------------------------------------------------------------------------- /AICare-baselines/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/requirements.txt -------------------------------------------------------------------------------- /AICare-baselines/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/test.py -------------------------------------------------------------------------------- /AICare-baselines/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/AICare-baselines/train.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/README.md -------------------------------------------------------------------------------- /config/config_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/config/config_example.py -------------------------------------------------------------------------------- /datasets/mimic-iv/extract_nshot.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/datasets/mimic-iv/extract_nshot.ipynb -------------------------------------------------------------------------------- /datasets/mimic-iv/preprocess_mimic4.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/datasets/mimic-iv/preprocess_mimic4.ipynb -------------------------------------------------------------------------------- /datasets/preprocess/statistics.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/preprocess/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/datasets/preprocess/tools.py -------------------------------------------------------------------------------- /datasets/tjh/extract_nshot.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/datasets/tjh/extract_nshot.ipynb -------------------------------------------------------------------------------- /datasets/tjh/preprocess_tjh.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/datasets/tjh/preprocess_tjh.ipynb -------------------------------------------------------------------------------- /datasets/tjh/promptgen_tjh.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/datasets/tjh/promptgen_tjh.ipynb -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/metrics/__init__.py -------------------------------------------------------------------------------- /metrics/binary_classification_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/metrics/binary_classification_metrics.py -------------------------------------------------------------------------------- /metrics/metrics_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/metrics/metrics_utils.py -------------------------------------------------------------------------------- /metrics/regression_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/metrics/regression_metrics.py -------------------------------------------------------------------------------- /performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/performance.py -------------------------------------------------------------------------------- /prompts/mimic-iv/1-1_example.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/mimic-iv/1-1_example.txt -------------------------------------------------------------------------------- /prompts/mimic-iv/N-1_batches_example.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/mimic-iv/N-1_batches_example.txt -------------------------------------------------------------------------------- /prompts/mimic-iv/N-1_list_example.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/mimic-iv/N-1_list_example.txt -------------------------------------------------------------------------------- /prompts/mimic-iv/range.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/mimic-iv/range.json -------------------------------------------------------------------------------- /prompts/mimic-iv/string_example.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/mimic-iv/string_example.txt -------------------------------------------------------------------------------- /prompts/mimic-iv/type.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/mimic-iv/type.json -------------------------------------------------------------------------------- /prompts/mimic-iv/unit.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/mimic-iv/unit.json -------------------------------------------------------------------------------- /prompts/prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/prompt.py -------------------------------------------------------------------------------- /prompts/template.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/template.txt -------------------------------------------------------------------------------- /prompts/tjh/1-1_example.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/tjh/1-1_example.txt -------------------------------------------------------------------------------- /prompts/tjh/N-1_batches_example.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/tjh/N-1_batches_example.txt -------------------------------------------------------------------------------- /prompts/tjh/N-1_list_example.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/tjh/N-1_list_example.txt -------------------------------------------------------------------------------- /prompts/tjh/range.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/tjh/range.json -------------------------------------------------------------------------------- /prompts/tjh/string_example.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/tjh/string_example.txt -------------------------------------------------------------------------------- /prompts/tjh/type.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/tjh/type.json -------------------------------------------------------------------------------- /prompts/tjh/unit.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/prompts/tjh/unit.json -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/llm4healthcare/HEAD/run.py --------------------------------------------------------------------------------