Skip to content

jaehunjung1/Hierarchical-Attention-Network

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits

best_model

best_model

 
 

data

data

 
 

.gitignore

.gitignore

 
 

README.md

README.md

 
 

dataloader.py

dataloader.py

 
 

dataset.py

dataset.py

 
 

model.py

model.py

 
 

prediction_bar_chart.png

prediction_bar_chart.png

 
 

result.html

result.html

 
 

test.py

test.py

 
 

train.py

train.py

 
 

trainer.py

trainer.py

 
 

utils.py

utils.py

 
 

vis_preview.png

vis_preview.png

 
 

Repository files navigation

Hierarchical-Attention-Network

HAN implemented in Pytorch, along with attention visualization

preview_img

usage
  • Download and add glove.6B.100d.txt into ./data/glove
  • Run python train.py <args>
  • Run python test.py to visualize attention

About

A PyTorch implementation of the document classification by Hierarchical Attention Network

Topics

document-classification hierarchical-attention-networks pytorch-implementation

Resources

Readme
Activity

Stars

32 stars

Watchers

2 watching

Forks

9 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages