Graph Convolutional Networks in PyTorch
Switch branches/tags
Nothing to show
Clone or download
tkipf Merge pull request #25 from wbadart/patch-1
models.py: absolute import layers.GraphConvolution
Latest commit 4396e4d Oct 20, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data/cora Initial commit Oct 26, 2017
pygcn models.py: absolute import layers.GraphConvolution Oct 17, 2018
.gitignore Initial commit Oct 26, 2017
LICENCE Initial commit Oct 26, 2017
README.md Update README.md Aug 5, 2018
figure.png Initial commit Oct 26, 2017
setup.py Initial commit Oct 26, 2017

README.md

Graph Convolutional Networks in PyTorch

PyTorch implementation of Graph Convolutional Networks (GCNs) for semi-supervised classification [1].

For a high-level introduction to GCNs, see:

Thomas Kipf, Graph Convolutional Networks (2016)

Graph Convolutional Networks

Note: There are subtle differences between the TensorFlow implementation in https://github.com/tkipf/gcn and this PyTorch re-implementation. This re-implementation serves as a proof of concept and is not intended for reproduction of the results reported in [1].

This implementation makes use of the Cora dataset from [2].

Installation

python setup.py install

Requirements

  • PyTorch 0.4 or 0.5
  • Python 2.7 or 3.6

Usage

python train.py

References

[1] Kipf & Welling, Semi-Supervised Classification with Graph Convolutional Networks, 2016

[2] Sen et al., Collective Classification in Network Data, AI Magazine 2008

Cite

Please cite our paper if you use this code in your own work:

@article{kipf2016semi,
  title={Semi-Supervised Classification with Graph Convolutional Networks},
  author={Kipf, Thomas N and Welling, Max},
  journal={arXiv preprint arXiv:1609.02907},
  year={2016}
}