• Stars
    star
    110
  • Rank 314,861 (Top 7 %)
  • Language
    Python
  • Created over 6 years ago
  • Updated almost 5 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

document classification using LSTM + self attention

document classification LSTM + self attention

Pytorch implementation of LSTM classification with self attention. See A STRUCTURED SELF-ATTENTIVE SENTENCE EMBEDDING

Some resutls -> my blog post

IMDB Experiments

training

python imdb_attn.py

visualize attention

python view_attn.py

results ./attn.html: label \t pred label \t sentence with attention(<span ....>)