Hierarchical Attentional Hybrid Neural Networks for Document Classification
J. Abreu , L. Fred, D. Macêdo, C. Zanchettin, "Hierarchical Attentional Hybrid Neural Networks for Document Classification", Submitted to IJCNN on 15 Jan, 2019.
Datasets:
Dataset | Classes | Documents | source |
---|---|---|---|
Yelp Review Polarity | 5 | 1569264 | link |
IMDb Movie Review | 2 | 50000 | link |
To download datasets, install the kaggle tool:
pip install kaggle
then run follow commands:
kaggle datasets download -d luisfredgs/in1164-deep-learning
kaggle datasets download -d luisfredgs/wiki-news-300d-1m-subword
put all data on input/ folder
Requirements
- tensorflow 1.10
- Keras 2.x
- gensim
- tqdm
- matplotlib
A GPU with CUDA support is required to run this code
Execution
run python train.py