/gae-pytorch

Graph Auto-Encoder in PyTorch

Primary LanguagePythonMIT LicenseMIT

gae-pytorch

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.

Requirements

  • Python 3
  • PyTorch 0.4
  • install requirements via pip install -r requirements.txt

How to run

python gae/train.py