-
-
Notifications
You must be signed in to change notification settings - Fork 110
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.
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
Visualize attention matrix in SAITS #302
Conversation
Pull Request Test Coverage Report for Build 10088161856Warning: This coverage report may be inaccurate.This pull request's base commit is no longer the HEAD commit of its target branch. This means it includes changes from outside the original pull request, including, potentially, unrelated coverage changes.
Details
💛 - Coveralls |
Many thanks, Anshu, for your first PR to PyPOTS! I've refactored your code by removing typos and unused imports. Could you please run your function visualize_attention() just after obtaining results from SAITS in this line of code here PyPOTS/tests/imputation/saits.py Line 67 in 3ff6887
so we can test and ensure the function works well |
This pull request had no activity for 14 days. It will be closed in 1 week unless there is some new activity. |
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
Hi Wenjie, I have made edits to the file to address errors in setting axis parameters. I have also made a change to return the Figure object as opposed to an Axes object and added an option for adjusting the font scale of the Seaborn heatmap. The function has been tested locally and works with the output of self.saits.predict(). Thank you! |
This pull request had no activity for 14 days. It will be closed in 1 week unless there is some new activity. |
This pull request had no activity for 14 days. It will be closed in 1 week unless there is some new activity. |
Added file visualizeAttention.py to pypots.utils.visual for visualization of attention matrix from DMSA heads of SAITS.
The required dependencies are: numpy, seaborn, matplotlib
Fixing #178