Visualization result implementation

Dear author, thank you very much for opening up this project. We are very interested in it and have conducted experiments to reproduce it. However, we found that no visualization results were generated in the end. Could you please open source this part of the code in your spare time for us to further learn and refer to? Thank you very much, looking forward to your reply!