xrosliang / pytorch-gat Goto Github PK
View Code? Open in Web Editor NEWThis project forked from gordicaleksa/pytorch-gat
My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I'll add an inductive example (PPI) soon as well.
Home Page: https://youtube.com/c/TheAIEpiphany
License: MIT License