pietramarrone / 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've supported both Cora (transductive) and PPI (inductive) examples!
Home Page: https://youtube.com/c/TheAIEpiphany
License: MIT License