Skip to content

liaoshiyun/self-attention-classification

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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 ....>)

About

document classification using LSTM + self attention

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 96.2%
  • Dockerfile 3.8%