MXNet implementation of Graph Convolutional Neural Networks detailed in Semi-Supervised Classification with Graph Convolutional Networks
Tensorflow Implementation (Original) Author's PyTorch Implementation Pytorch Implementation by Bumsoo Kim
Graph Convolutions are best explained in this amazing blogpost by Thomas Kipf the author of the paper Semi-Supervised Classification with Graph Convolutional Networks
- Python
- MXnet
- Numpy
python train.py
- Unable to reproduce results from the paper with dropout of 0.5. Dropout=0 gives results similar to paper
- NetworkX integration for loading data
- Add options to pick loss function
- Support datasets other than CORA
- Option to save model
- Add visualization support of loss parameters