# Learning Deep Generative Models of Graphs

This is an implementation of [Learning Deep Generative Models of Graphs](https://arxiv.org/pdf/1803.03324.pdf) by 
Yujia Li, Oriol Vinyals, Chris Dyer, Razvan Pascanu, Peter Battaglia. 

## Dependency
- Python 3.5.2
- [Pytorch 0.4.1](https://pytorch.org/)
- [Matplotlib 2.2.2](https://matplotlib.org/)

## Usage

- Train with batch size 1: `python main.py`
- Train with batch size larger than 1: `python main_batch.py`.

## Acknowledgement

We would like to thank Yujia Li for providing details on the implementation.