https://github.com/limaosen0/Variational-Graph-Auto-Encoders

This is the implementation of paper 'Variational Graph Auto-Encoder' in NIPS Workshop on Bayesian Deep Learning, 2016.
https://github.com/limaosen0/Variational-Graph-Auto-Encoders

Last synced: 3 months ago
JSON representation

This is the implementation of paper 'Variational Graph Auto-Encoder' in NIPS Workshop on Bayesian Deep Learning, 2016.

• Host: GitHub
• URL: https://github.com/limaosen0/Variational-Graph-Auto-Encoders
• Owner: limaosen0
• Created: 2018-05-17T07:56:32.000Z (over 6 years ago)
• Default Branch: master
• Last Pushed: 2018-05-20T13:58:25.000Z (over 6 years ago)
• Last Synced: 2024-05-01T01:33:20.524Z (6 months ago)
• Language: Python
• Homepage:
• Size: 314 KB
• Stars: 72
• Watchers: 2
• Forks: 23
• Open Issues: 2
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• graph-networks - Variational Graph Auto-Encoder
• StarryDivineSky - Variational-Graph-Auto-Encoders

README

        
# Variational-Graph-Auto-Encoders

This is the implementation of paper "Variational Graph Auto-Encoders", which is published in NIPS 2016 Workshop.

Thomas N. Kipf, Max Welling, Variational Graph Auto-Encoders, In NIPS Workshop on Bayesian Deep Learning, 2016.

# How to run the code?

Insure that you have 4 GB memory in your GPU and you have installed the required module.

We have provided the training data in right form in './data', and the main program 'main.py' can be run directly, like input 'CUDA_VISIBLE_DEVICES=0 python main.py' in terminal. For the first time of running, the program will load the data and generate a graph, which might cost much time. After that, a loaded graph would be saved in a numpy file in './data', and you can load the graph more efficiently.

During the training process, the program would print the loss value and validation accuracy. At last, the ROC curve would be saved in './result'.

# Result

The result of our program is shown here (for citation dataset and facebook dataset, respectively).

1 Citation

![ROC_curve_citation](https://github.com/limaosen0/Variational-Graph-Auto-Encoders/blob/master/result/ROC_curve_citation_.png)

2 Facebook

![ROC_curve_facebook](https://github.com/limaosen0/Variational-Graph-Auto-Encoders/blob/master/result/ROC_curve_facebook_.png)

# Tools and Modules

Tensorflow 1.5

igraph-python

numpy, scipy.sparse and matplotlib

# Notes

You can clone our program for practice, but if you use it in your published paper, please cite the paper "Variational Graph Auto-Encoders".