HiTANet: Hierarchical Time-Aware Attention Networks for Risk Prediction on Electronic Health Records
This is the code of HiTANet paper HiTANet: Hierarchical Time-Aware Attention Networks for Risk Prediction on Electronic Health Records https://www.kdd.org/kdd2020/accepted-papers/view/hitanet-hierarchical-time-aware-attention-networks-for-risk-prediction-on-e
pytorch == 1.3.1 sklearn == 0.19.1
The project contains a sample toy data named as hf_sample in data folder.
train_hitanet.py contains training and evaluation codes.