"vscode:/vscode.git/clone" did not exist on "c2fbf60f398c1d44b675c0fc3fbd3039dca01003"
README.md 772 Bytes
Newer Older
1
2
3
# 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 
Mufei Li's avatar
Mufei Li committed
4
5
6
Yujia Li, Oriol Vinyals, Chris Dyer, Razvan Pascanu, Peter Battaglia.

For molecule generation, see 
Mufei Li's avatar
Mufei Li committed
7
[DGL-LifeSci](https://github.com/awslabs/dgl-lifesci/tree/master/examples/generative_models/dgmg).
8

9
## Dependencies
10
11
12
13
- 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
14
## Usage
15

Mufei Li's avatar
Mufei Li committed
16
`python3 main.py`
Mufei Li's avatar
Mufei Li committed
17

18
19
20
21
22
23
## Performance

90% accuracy for cycles compared with 84% accuracy reported in the original paper.

## Speed

Mufei Li's avatar
Mufei Li committed
24
On AWS p3.2x instance (w/ V100), one epoch takes ~526s.
25

Mufei Li's avatar
Mufei Li committed
26
27
28
## Acknowledgement

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