Graph Auto-Encoder in PyTorch
This is a PyTorch implementation of the Variational Graph Auto-Encoder model described in the paper:
T. N. Kipf, M. Welling, Variational Graph Auto-Encoders, NIPS Workshop on Bayesian Deep Learning (2016)
The code in this repo is based on or refers to https://github.com/tkipf/gae, https://github.com/tkipf/pygcn and https://github.com/vmasrani/gae_in_pytorch.
- Python 3
- PyTorch 0.4
- install requirements via
pip install -r requirements.txt
python train.py