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!

语言：Jupyter Notebook · GitHub 星标：2,691 · Fork：356

话题：attention · attention-mechanism · deep-learning · gat · gat-tutorial · graph-attention-network · graph-attention-networks · jupyter

在 GitHub 上查看原仓库