Skip to content

This repository implements variational graph auto encoder by Thomas Kipf.

License

Notifications You must be signed in to change notification settings

DaehanKim/vgae_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Variational Graph Auto-encoder in Pytorch

This repository implements variational graph auto-encoder by Thomas Kipf. For details of the model, refer to his original tensorflow implementation and his paper.

Requirements

  • Pytorch
  • python 3.x
  • networkx
  • scikit-learn
  • scipy

How to run

  • Specify your arguments in args.py : you can change dataset and other arguments there
  • run python train.py

Notes

  • The dataset is the same as what Kipf provided in his original implementation. Thus I used his preprocessing code as-is(maybe with minor modification).
  • Per-epoch training time is a bit slower then the original implementation.(0.2 sec/epoch --> 0.9 sec/epoch)
  • Train accuracy, validation(test) average precision, auroc are similar to those of the original. (over 90% for both AP and roc)
  • Dropout is not implemented now.
  • Feel free to report some inefficiencies in the code! (It's just initial version so may have much room for pytorch-adaptation)

About

This repository implements variational graph auto encoder by Thomas Kipf.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages