WangliLin / SAH-RNN

The implementation of the Structure-Aware Hierarchical LSTM.

Source Code of the Structure-Aware Hierarchical LSTM.

root_path
├── layers.py: the definition of Hierarchical LSTM Layer and Dual Attention Layer
├── models.py: the definition of Structure-Aware Hierarchical LSTM model
├── test.py: code for model predicting
├── train.py: code for model training
└── utils.py: some functions like R@P=0.1 and AUC here