Skip to content

alexarnimueller/MolGAN

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MolGAN

Tensorflow implementation of MolGAN: An implicit generative model for small molecular graphs (https://arxiv.org/abs/1805.11973)

Overview

This library contains a Tensorflow implementation of MolGAN: An implicit generative model for small molecular graphs as presented in [1](https://arxiv.org/abs/1805.11973).

Dependencies

Structure

  • data: should contain your datasets. If you run download_dataset.sh the script will download the dataset used for the paper (then you should run utils/sparse_molecular_dataset.py to convert the dataset in a graph format used by MolGAN models).
  • example: Example code for using the library within a Tensorflow project. NOTE: these are NOT the experiments on the paper!
  • models: Class for Models. Both VAE and (W)GAN are implemented.
  • optimizers: Class for Optimizers for both VAE, (W)GAN and RL.

Usage

Please have a look at the example.

Please cite [1] in your work when using this library in your experiments.

Feedback

For questions and comments, feel free to contact Nicola De Cao.

License

MIT

Citation

[1] De Cao, N., and Kipf, T. (2018).MolGAN: An implicit generative 
model for small molecular graphs. ICML 2018 workshop on Theoretical
Foundations and Applications of Deep Generative Models.

BibTeX format:

@article{de2018molgan,
  title={{MolGAN: An implicit generative model for small
  molecular graphs}},
  author={De Cao, Nicola and Kipf, Thomas},
  journal={ICML 2018 workshop on Theoretical Foundations 
  and Applications of Deep Generative Models},
  year={2018}
}

About

Tensorflow implementation of MolGAN: An implicit generative model for small molecular graphs (https://arxiv.org/abs/1805.11973)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.7%
  • Shell 0.3%