项目作者: nn116003

项目描述 :
document classification using LSTM + self attention
高级语言: Python
项目地址: git://github.com/nn116003/self-attention-classification.git
创建时间: 2018-02-08T14:10:08Z
项目社区:https://github.com/nn116003/self-attention-classification

开源协议:

下载


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

  1. python imdb_attn.py

visualize attention

  1. python view_attn.py

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