Skip to content
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

Add attention map visualization func #475

Merged
merged 4 commits into from
Jul 25, 2024
Merged

Add attention map visualization func #475

merged 4 commits into from
Jul 25, 2024

Commits on Jul 22, 2024

  1. refactor: code refactoring;

    @WenjieDu
    WenjieDu committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    00a45f5 View commit details
    Browse the repository at this point in the history

  2. fix: unclosed '(';

    @WenjieDu
    WenjieDu committed Jul 22, 2024
    Configuration menu
    Copy the full SHA
    fba4a5e View commit details
    Browse the repository at this point in the history

Commits on Jul 25, 2024

  1. Visualize attention matrix in SAITS (#302) 

    * Update visualizeAttention.py

Changed typos leading to errors in the function. Returns the figure object instead of axis object. Added a parameter for setting the font scale in the heatmap

* Update visualizeAttention.py

---------

Co-authored-by: Wenjie Du <wenjay.du@gmail.com>
    @gugababa @WenjieDu
    gugababa and WenjieDu authored Jul 25, 2024
    Configuration menu
    Copy the full SHA
    b655637 View commit details
    Browse the repository at this point in the history

  2. refactor: rename into attention_map;

    @WenjieDu
    WenjieDu committed Jul 25, 2024
    Configuration menu
    Copy the full SHA
    2ddd288 View commit details
    Browse the repository at this point in the history