Adding an attention layer to the multiclass classification

utterworks / fast-bert

Super easy library for BERT based NLP models

Apache License 2.0

1.85k stars 342 forks source link

Closed Ting-Kerbell closed 3 years ago

Ting-Kerbell commented 3 years ago

How can I add an attention layer to the multiclass classification, so that I can see what words weigh more in the model's decision to classify?