How to get attention weights vector in Han?

ShawnyXiao / TextClassification-Keras

Text classification models implemented in Keras, including: FastText, TextCNN, TextRNN, TextBiRNN, TextAttBiRNN, HAN, RCNN, RCNNVariant, etc.

MIT License

816 stars 187 forks source link

How to get attention weights vector in Han? #12

Open rahmanjalayer opened 3 years ago

rahmanjalayer commented 3 years ago

Hi, Thank you for these codes. I wonder if it is possible to get attention weights vector on word-level and also sentence-level attention in Han model? I want to plot these weights vector after predicting its category.