README.md 563 Bytes
Newer Older
1
2
3
4
5
# 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. 

Mufei Li's avatar
Mufei Li committed
6
## Dependency
7
8
9
10
- Python 3.5.2
- [Pytorch 0.4.1](https://pytorch.org/)
- [Matplotlib 2.2.2](https://matplotlib.org/)

Mufei Li's avatar
Mufei Li committed
11
## Usage
12
13

- Train with batch size 1: `python main.py`
Mufei Li's avatar
Mufei Li committed
14
15
16
17
18
- 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.