microsoft/rat-sql

How did you generate the heatmap plots for the attentions?

Opened this issue · 2 comments

senthurRam33 commented

In enc_dec.py in class EncDecModel and method begin_inference you have a variable enc_state that holds the heatmap data I think. Not so sure.

If you plot that you will get something like this:
sns

manzambi11 commented

@senthurRam33 can you share you code please