Graphite: Iterative Generative Modeling of Graphs
Source code for our paper "Graphite: Iterative Generative Modeling of Graphs".
If you find it helpful, please consider citing our paper.
@article{grover2018iterative,
title={Graphite: Iterative Generative Modeling of Graphs},
author={Grover, Aditya and Zweig, Aaron and Ermon, Stefano},
journal={arXiv preprint arXiv:1803.10459},
year={2018}
}
Requirements
- python 2.7
- Tensorflow
- Networkx 1.11
Training
python train.py --epochs 500 --model feedback --edge_dropout 0.5 --learning_rate 0.01 --autoregressive_scalar 0.5
This code was built on top of a graph autoencoder (GAE) implementation available here.
If you have any questions, feel free to contact [email protected] or [email protected].