Skip to content

ying-he/Attention-GRU_text_classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 

Repository files navigation

Attention-GRU_text_classification

Codes in this repository is an implement of HAN model, which is based on GRU network and attention mechanism, and the original paper is "Implementation of Hierarchical Attention Networks for Document Classification" (Yang et al., 2017).

I use Amazon Review datasets to do this implementation.

Version requirements:

  • Python 3.6

  • Tensorflow 1.12+

  • numpy

  • gensim (for training word2vec model and obtaining word embeddings)

And my codes are referring to Irvinglove's codes.

Thank to all.

About

Use HAN model, which is based on GRU network and attention mechanism, to do long-text classification.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages