How did you generate the heatmap plots for the attentions?

microsoft / rat-sql

A relation-aware semantic parsing model from English to SQL

https://arxiv.org/abs/1911.04942

MIT License

406 stars 117 forks source link

Open JoaoLages opened 3 years ago

JoaoLages commented 3 years ago

senthurRam33 commented 3 years ago

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 2 years ago

@senthurRam33 can you share you code please