gangeshwark / hierarchical_document_modeling

Hierarchical Attention Networks for Document Classification

Hierarchical Attention Networks for Document Classification

Implementation of hierarchical attention model described in Hierarchical Attention Networks for Document Classification (Yang et al., 2016).

Run

python train.py --epochs 30 --batch_size 32 --lr 0.001

Credits: Used part of code from this repo .