This is a simple Graph Auto-Encoder [1] implemented with DGL.
You can directly open the 66_train.ipynb to test the model.
You can switch the dataset to use from cora/citeseer/pubmed.
- pytorch
- dgl
- networkx
- numpy
- scikit-learn
- scipy
[1] Thomas N. Kipf and Max Welling. "Variational Graph Auto-Encoders." NIPS. 2016.
[2] github.com/DaehanKim/vgae_pytorch