luisfredgs / Cnn Hierarchical Network For Document Classification
Labels
Projects that are alternatives of or similar to Cnn Hierarchical Network For Document Classification
Hierarchical Attentional Hybrid Neural Networks for Document Classification
This paper was accepted in ICANN 2019
J. Abreu , L. Fred, D. Macêdo, C. Zanchettin, "Hierarchical Attentional Hybrid Neural Networks for Document Classification".
Performance on Yelp Dataset multi-class
Datasets:
Dataset | Classes | Documents | download |
---|---|---|---|
Yelp Review Polarity | 5 | 1569264 | link |
IMDb Movie Review | 2 | 50000 | link |
Do you want use Pre-trained FastText word embeddings? Downloaded in https://www.kaggle.com/luisfredgs/wiki-news-300d-1m-subword. Check the source code for more details. Pay attention to Colab limits of RAM and GPU.
Requirements
- Python 3
- tensorflow 1.10
- Keras 2.x
- spacy 2.0
- gensim
- tqdm
- matplotlib
A GPU with CUDA support is required to run this code.
Run this code on Google Colab with Free GPU
On Google Colab, Select "Runtime," "Change runtime type" to Python 3. Ensure "Hardware accelerator" is set to GPU (the default is CPU).
To run this notebook on Google Colab you don't need download dataset files. Type your kaggle username and API key during cell execution and wait. Will done. If do you want to make predictions on new text data using a trained model, check make_predictions.ipynb for more details.
Please cite
@article{abreu2019hierarchical,
title={Hierarchical Attentional Hybrid Neural Networks for Document Classification},
author={Abreu, Jader and Fred, Luis and Mac{\^e}do, David and Zanchettin, Cleber},
journal={arXiv preprint arXiv:1901.06610},
year={2019}
}