nn116003/self-attention-classification

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

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

training

python imdb_attn.py

visualize attention

python view_attn.py

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

torchtext-tutorial

End-to-end-Learning-of-Deterministic-Decision-Trees

allennlp-ja-tutorial