Skip to content

Latest commit

 

History

History
29 lines (26 loc) · 958 Bytes

README.md

File metadata and controls

29 lines (26 loc) · 958 Bytes

graphvae_approx

Tensorflow implementation of the model described in the paper Efficient Learning of Non-Autoregressive Graph Variational Autoencoders for Molecular Graph Generation

Components

  • preprocessing.py - script for preprocessing data
  • train.py - script for model training
  • test.py - script for model evaluation (molecular graph generation)
  • GVAE.py - model architecture

Dependencies

  • Python
  • TensorFlow
  • RDKit
  • NumPy
  • scikit-learn
  • sparse

Citation

@Article{Kwon2019,
  title={Efficient learning of non-autoregressive graph variational autoencoders for molecular graph generation},
  author={Kwon, Youngchun and Yoo, Jiho and Choi, Youn-Suk and Son, Won-Joon and Lee, Dongseon and Kang, Seokho},
  journal={Journal of Cheminformatics},
  volume={11},
  pages={70},
  year={2019},
  doi={10.1186/s13321-019-0396-x}
}