← all repositories

PetarV-/GAT

A TensorFlow implementation of Graph Attention Networks (GAT), a graph neural network layer that uses attention mechanisms for node classification tasks.

3.5k stars Python ML Frameworks
GAT
Velocity · 7d
+1.2
★ / day
Trend
steady
star history

This repository provides a reference implementation of the Graph Attention Network paper (ICLR 2018), featuring both dense and experimental sparse versions of the GAT layer in TensorFlow. It includes utilities for preprocessing the Cora dataset and PPI benchmark, along with a pretrained model achieving 84.4% accuracy on Cora’s test set. The implementation handles graph structure via adjacency matrices and uses multi-head attention for node representation learning.

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