How to visualize the learned Attention?

Diego999 / pyGAT

Pytorch implementation of the Graph Attention Network model by Veličković et. al (2017, https://arxiv.org/abs/1710.10903)

MIT License

2.92k stars 692 forks source link

Closed SuperBruceJia closed 3 years ago

SuperBruceJia commented 3 years ago

Dear Diego,

Thanks a lot for your great work! It helps me a lot!

One thing I wanna inquire about is that for the GAT model, is there a way to visualize the Attention learned by the network?

Thanks in advance and have a nice day!

Best,

Shuyue