How to visualize attention?

Support
1

Hello,

How to go about visualizing attention using OpenNMT-py?

1 Like
2

Figured some way around myself.

I used Translation class and logged everything required - src, tgt, pred, attentions into a file corresponding to the batches, which worked enough for me. Once you have (src, pred, attention matrix), it’s pretty straightforward.

1 Like
3

Thanks for your sharing, it helps me a lot!