q20110911 / core

CORE: Automatic Molecule Optimization using Copy & Refine Strategy (AAAI 2020)

CORE: Automatic Molecule Optimization using Copy and Refine Strategy (AAAI 2020)

This repo relies heavily on Graph2Graph (https://github.com/wengong-jin/iclr19-graph2graph).

Requirements (Following Graph2Graph)

• Python == 2.7
• RDKit >= 2017.09
• PyTorch >= 0.4.0
• Numpy
• scikit-learn

The code has been tested under python 2.7 with pytorch 0.4.1.

Quick Start (Following Graph2Graph)

The tutorial of training and testing our variational junction tree encoder-decoder is in diff_vae/README.md.

A quick summary of different folders:

• data/ contains the training, validation and test set of logP, QED and DRD2 tasks described in the paper.
• fast_jtnn/ contains the implementation of junction tree encoder-decoder.
• diff_vae/ includes the training and decoding script of variational junction tree encoder-decoder (README).
• diff_vae_gan/ includes the training and decoding script of adversarial training module (README).
• props/ is the property evaluation module, including penalized logP, QED and DRD2 property calculation.
• scripts/ provides evaluation and data preprocessing scripts.

Implementation of CORE

• fast_jtnn/xxx.py

Contact

Tianfan Fu ([email protected])