New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

How did you generate the heatmap plots for the attentions? #53

Open

JoaoLages opened this issue May 21, 2021 · 2 comments

JoaoLages commented May 21, 2021

senthurRam33 commented Jun 1, 2021

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:

manzambi11 commented Aug 17, 2022

@senthurRam33 can you share you code please

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment