← all repositories

diegoantognini/pyGAT

PyTorch implementation of the Graph Attention Network (GAT) model for graph-structured data learning.

3.1k stars Python ML Frameworks
pyGAT
Velocity · 7d
+1.0
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of the Graph Attention Network architecture introduced by Veličković et al. (2018). The GAT applies self-attention mechanisms to graph nodes to learn importance weights between connected vertices, enabling inductive learning on graph data. It includes training code for transductive learning tasks (Cora dataset), benchmark performance metrics, and both dense and sparse matrix implementations. The project is forked from the classic PyGCN repository and references the original TensorFlow implementation.

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