Giters

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

gordicaleksa/pytorch-GAT Stargazers