CS224W - Bag of Tricks for Node Classification with GNN - GAT Normalization#9840

Open

liuvince wants to merge 16 commits intopyg-team:masterfrom mattjhayes3:gat_normalization

+367-11

Commits on Dec 11, 2024

Use unnormalized attention matrix for GAT

Vincenzooo
authored and
Vincenzooo
committed
Update GAT norm with torch_csr_tensor

Vincenzooo
authored and
Vincenzooo
committed
Add full test to GAT Conv with normalization

Vincenzooo
authored and
Vincenzooo
committed
Add normalization to GATv2Conv

Vincenzooo
authored and
Vincenzooo
committed
Add GAT normalization parameter in benchmark/citation/gat.py

Vincenzooo
authored and
Vincenzooo
committed
Fix typo

Vincenzooo
authored and
Vincenzooo
committed
Remove unused variable and passing more parameters to gat_norm

Vincenzooo
authored and
Vincenzooo
committed
Outline the fact that GAT Normalization does not handle bipartite message passing yet and add num_nodes as parameters

Vincenzooo
authored and
Vincenzooo
committed
Checking self.normalize once

Vincenzooo
authored and
Vincenzooo
committed
Update test condition to check bipartite graph in forward pass

Vincenzooo
authored and
Vincenzooo
committed
Fix gat_norm when edge_index is SparseTensor

Vincenzooo
authored and
Vincenzooo
committed
Add small fixes

Vincenzooo
authored and
Vincenzooo
committed
Clean typing and add typing test

Vincenzooo
authored and
Vincenzooo
committed
Update CHANGELOG.md

Vincenzooo
authored and
Vincenzooo
committed
Update tests

Vincenzooo
authored and
Vincenzooo
committed

Commits on Dec 13, 2024

Update flow

Vincenzooo
authored and
Vincenzooo
committed