Diff of /README.md [000000] .. [9e1f38]

Switch to unified view

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
![Screenshot](screenshot.png)
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/>