← all repositories

gordicaleksa/pytorch-GAT

A PyTorch implementation of the Graph Attention Network (GAT) architecture from Veličković et al. with Cora and PPI dataset examples.

2.7k stars Jupyter Notebook ML FrameworksOther AI
pytorch-GAT
Velocity · 7d
+1.3
★ / day
Trend
steady
star history

This repository provides a clean implementation of Graph Attention Networks in PyTorch. GATs apply self-attention mechanisms to graph-structured data, enabling nodes to weight their neighbors’ features adaptively. The implementation includes transductive learning on the Cora dataset and inductive learning on the PPI dataset, along with visualization tools for t-SNE embeddings, attention patterns, and entropy histograms.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.