/GF-VAE

A Flow-based Variational Autoencoder for Molecule Generation

Primary LanguageJupyter Notebook

GF-VAE

A Flow-based Variational Autoencoder for Molecule Generation

the code can be separated into two parts, skeleton of the flow part is from paper MoFlow, which you may approach at https://github.com/calvin-zcx/moflow, thanks for his great job. The detailed code is shown in model folder.