Skip to content

Latest commit

 

History

History
31 lines (23 loc) · 938 Bytes

README.md

File metadata and controls

31 lines (23 loc) · 938 Bytes
Raw

Graph Convolutional Networks

A TensorFlow 2 implementation of Graph Convolutional Networks for classification of nodes from the paper, Thomas N. Kipf, Max Welling, Semi-Supervised Classification with Graph Convolutional Networks (ICLR 2017)

This is my attempt at trying to understand and recreate the neural network from from the paper. You can find the official implementation here: https://github.com/tkipf/gcn

Requirements

  • tensorflow 2
  • networkx
  • numpy
  • scikit-learn

Run

To train and test the network with the CORA dataset.

python train.py

Cite

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

@inproceedings{kipf2017semi,
  title={Semi-Supervised Classification with Graph Convolutional Networks},
  author={Kipf, Thomas N. and Welling, Max},
  booktitle={International Conference on Learning Representations (ICLR)},
  year={2017}
}