├── .gitignore ├── LICENSE ├── README.md ├── analyze.py ├── configs ├── config.yaml ├── dl.py ├── experiments.py └── ml.py ├── datasets ├── cdsl │ └── raw │ │ └── .gitkeep ├── loader │ ├── datamodule.py │ ├── load_los_info.py │ └── unpad.py ├── preprocess │ ├── statistics.py │ └── tools.py ├── preprocess_cdsl.ipynb ├── preprocess_tjh.ipynb └── tjh │ └── raw │ └── .gitkeep ├── dl_tune.py ├── losses ├── __init__.py ├── multitask_loss.py └── time_aware_loss.py ├── metrics ├── __init__.py ├── binary_classification_metrics.py ├── es.py ├── fairness_metrics.py ├── osmae.py ├── regression_metrics.py └── utils.py ├── ml_tune.py ├── models ├── __init__.py ├── adacare.py ├── agent.py ├── catboost.py ├── concare.py ├── dt.py ├── gbdt.py ├── grasp.py ├── gru.py ├── heads.py ├── lstm.py ├── mcgru.py ├── mlp.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 ├── plot.ipynb ├── requirements.txt ├── test.py ├── test_twostage.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/README.md -------------------------------------------------------------------------------- /analyze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/analyze.py -------------------------------------------------------------------------------- /configs/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/configs/config.yaml -------------------------------------------------------------------------------- /configs/dl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/configs/dl.py -------------------------------------------------------------------------------- /configs/experiments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/configs/experiments.py -------------------------------------------------------------------------------- /configs/ml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/configs/ml.py -------------------------------------------------------------------------------- /datasets/cdsl/raw/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/loader/datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/datasets/loader/datamodule.py -------------------------------------------------------------------------------- /datasets/loader/load_los_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/datasets/loader/load_los_info.py -------------------------------------------------------------------------------- /datasets/loader/unpad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/datasets/loader/unpad.py -------------------------------------------------------------------------------- /datasets/preprocess/statistics.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/preprocess/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/datasets/preprocess/tools.py -------------------------------------------------------------------------------- /datasets/preprocess_cdsl.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/datasets/preprocess_cdsl.ipynb -------------------------------------------------------------------------------- /datasets/preprocess_tjh.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/datasets/preprocess_tjh.ipynb -------------------------------------------------------------------------------- /datasets/tjh/raw/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dl_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/dl_tune.py -------------------------------------------------------------------------------- /losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/losses/__init__.py -------------------------------------------------------------------------------- /losses/multitask_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/losses/multitask_loss.py -------------------------------------------------------------------------------- /losses/time_aware_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/losses/time_aware_loss.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/metrics/__init__.py -------------------------------------------------------------------------------- /metrics/binary_classification_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/metrics/binary_classification_metrics.py -------------------------------------------------------------------------------- /metrics/es.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/metrics/es.py -------------------------------------------------------------------------------- /metrics/fairness_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/metrics/fairness_metrics.py -------------------------------------------------------------------------------- /metrics/osmae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/metrics/osmae.py -------------------------------------------------------------------------------- /metrics/regression_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/metrics/regression_metrics.py -------------------------------------------------------------------------------- /metrics/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/metrics/utils.py -------------------------------------------------------------------------------- /ml_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/ml_tune.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/adacare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/adacare.py -------------------------------------------------------------------------------- /models/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/agent.py -------------------------------------------------------------------------------- /models/catboost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/catboost.py -------------------------------------------------------------------------------- /models/concare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/concare.py -------------------------------------------------------------------------------- /models/dt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/dt.py -------------------------------------------------------------------------------- /models/gbdt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/gbdt.py -------------------------------------------------------------------------------- /models/grasp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/grasp.py -------------------------------------------------------------------------------- /models/gru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/gru.py -------------------------------------------------------------------------------- /models/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/heads.py -------------------------------------------------------------------------------- /models/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/lstm.py -------------------------------------------------------------------------------- /models/mcgru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/mcgru.py -------------------------------------------------------------------------------- /models/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/mlp.py -------------------------------------------------------------------------------- /models/retain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/retain.py -------------------------------------------------------------------------------- /models/rf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/rf.py -------------------------------------------------------------------------------- /models/rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/rnn.py -------------------------------------------------------------------------------- /models/stagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/stagenet.py -------------------------------------------------------------------------------- /models/tcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/tcn.py -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/transformer.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/utils.py -------------------------------------------------------------------------------- /models/xgboost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/models/xgboost.py -------------------------------------------------------------------------------- /pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/pipelines/__init__.py -------------------------------------------------------------------------------- /pipelines/dl_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/pipelines/dl_pipeline.py -------------------------------------------------------------------------------- /pipelines/ml_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/pipelines/ml_pipeline.py -------------------------------------------------------------------------------- /plot.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/plot.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/test.py -------------------------------------------------------------------------------- /test_twostage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/test_twostage.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yhzhu99/pyehr/HEAD/train.py --------------------------------------------------------------------------------