/HAN-text-classification

Primary LanguagePython

HAN-text-classification

This code belongs to the "Implementation of Hierarchical Attention Networks for Document Classification" .

Structure:

  • embedding

  • Word Encoder: word level bi-directional GRU to get rich representation of words

  • Word Attention:word level attention to get important information in a sentence

  • Sentence Encoder: sentence level bi-directional GRU to get rich representation of sentences

  • Sentence Attetion: sentence level attention to get important sentence among sentences

  • FC+Softmax

Requirement:

  • Python 2.7
  • Tensorflow 1.1 +
  • numpy

Notes:

Please load the dataset in this address. And add the dataset to data folder.

Hava a good time!