Chacha-Chen / UNITE

UNITE implementations

UNITE

UNITE implementations

Citation

[WWW' 21] UNITE: Uncertainty-based Health Risk Prediction Leveraging Multi-sourced Data Chacha Chen, Junjie Liang, Fenglong Ma, Lucas Glass, Jimeng Sun and Cao Xiao The World Wide Web Conference (WWW' 21)

Requirements

• Python 3.6x
• Pytorch 1.7.1

Files

• data/ contains your data in the ICD9 code format.
• CDC-data/ open-sourced demographics and geological data collected from CDC website
• UNITE.py: main running file
• baseline.py: implementation of the spatial temporal encoder

How to run

python UNITE.py # To run UNITE