|
a |
|
b/README.md |
|
|
1 |
# Targeted BEHRT |
|
|
2 |
Repository for publication: Targeted-BEHRT: Deep Learning for Observational Causal Inference on Longitudinal Electronic Health Records<br/> |
|
|
3 |
IEEE Transactions on Neural Networks and Learning Systems; Special Issue on Causality<br/> |
|
|
4 |
https://ieeexplore.ieee.org/document/9804397/<br/> |
|
|
5 |
DOI: 10.1109/TNNLS.2022.3183864.<br/> |
|
|
6 |
|
|
|
7 |
 |
|
|
8 |
|
|
|
9 |
How to use:<br/> |
|
|
10 |
In "examples" folder, run the "run_TBEHRT.ipynb" file. A test.csv file is provided to test/play and demonstrate how the vocabulary/year/age/etc function (please read full paper linked above for further methodological details). <br/> |
|
|
11 |
Furthermoree, in the examples folder to run the CV-TMLE estimator, run the "CVTMLE_example.ipynb" file. A host of fake fold data is provided to test/play and demonstrate how the CV-TMLE algorithm works (please read methods publication of CV-TMLE for further details). <br/> |
|
|
12 |
|
|
|
13 |
The files in the "src" folder contain model and data handling packages in addition to other necessary VAE relevant files and helper functions. |
|
|
14 |
|
|
|
15 |
Requirements:<br/> |
|
|
16 |
torch >1.6.0<br/> |
|
|
17 |
numpy 1.19.2<br/> |
|
|
18 |
sklearn 0.23.2<br/> |
|
|
19 |
pandas 1.1.3<br/> |
|
|
20 |
<br/> |