Skip to content

Tensorflow implementation of attention mechanism for text classification tasks.

License

Notifications You must be signed in to change notification settings

charlesfufu/tf-rnn-attention

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

55 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Tensorflow implementation of attention mechanism for text classification tasks.
Inspired by "Hierarchical Attention Networks for Document Classification", Zichao Yang et al. (http://www.aclweb.org/anthology/N16-1174).

Requirements

  • Python >= 2.6
  • Tensorflow >= 1.0
  • Keras (IMDB dataset)
  • tqdm

To view visualization example visit http://htmlpreview.github.io/?https://github.com/ilivans/tf-rnn-attention/blob/master/visualization.html

My bachelor's thesis on sentiment classification of Russian texts using Bi-RNN with attention mechanism: https://github.com/ilivans/attention-sentiment

About

Tensorflow implementation of attention mechanism for text classification tasks.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 73.4%
  • HTML 26.6%