项目作者: raunakkmr

项目描述 :
PyTorch implementation of Graph Attention Networks
高级语言: Python
项目地址: git://github.com/raunakkmr/Graph-Attention-Networks.git
创建时间: 2019-06-29T18:12:36Z
项目社区:https://github.com/raunakkmr/Graph-Attention-Networks

开源协议:MIT License

下载


Graph Attention Networks

This is a PyTorch implementation of Graph Attention Networks (GAT) from the paper Graph Attention Networks.

Usage

In the src directory, edit the config.json file to specify arguments and
flags. Then run python main.py.

Limitations

  • 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.

References