Need help with gae-pytorch?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

zfjsail
135 Stars 40 Forks MIT License 11 Commits 3 Opened issues

Description

Graph Auto-Encoder in PyTorch

Services available

!
?

Need anything else?

Contributors list

No Data

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/gaeinpytorch.

Requirements

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

How to run

python gae/train.py

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.