hecongqing / hierarchical-attention-networks

Implementation of Hierarchical Attention Networks as presented in https://www.cs.cmu.edu/~diyiy/docs/naacl16.pdf

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

hierarchical-attention-networks

Implementation of Hierarchical Attention Networks as presented in https://www.cs.cmu.edu/~diyiy/docs/naacl16.pdf

imdb_train.py trains the model with the IMDB sentiment data.

The input data for the model is ( batch docs, sentence, words). IMDB data contains a "sentence" per document, so we had an extra dimension as sentence.

About

Implementation of Hierarchical Attention Networks as presented in https://www.cs.cmu.edu/~diyiy/docs/naacl16.pdf

License:BSD 3-Clause "New" or "Revised" License


Languages

Language:Python 100.0%