PyTorch implementation of Graph Attention Networks.
This is a PyTorch implementation of Graph Attention Networks (GAT) from the paper Graph Attention Networks.
In the src
directory, edit the config.json
file to specify arguments and
flags. Then run python main.py
.
- Currently, only supports the Cora dataset. However, for a new dataset it should be fairly straightforward to write a Dataset class similar to
datasets.Cora
.
- Graph Attention Networks, Velickovic et al., ICLR 2018.
- Collective Classification in Network Data, Sen et al., AI Magazine 2008.