This is the code for the paper on https://www.biorxiv.org/content/10.1101/2020.05.23.112201v2
This repo introduces two things :
- A new Variational Auto-Encoder (VAE) architecture that goes from a molecular graph to a sequence representation (and especially SELFIEs).
- An optimization pipeline to iteratively shift a prior distribution to maximize a black-box scoring function. In our implemntation, this back-box function is the score returned by a docking software.
The necessary packages are packaged as yml files available for cpu or cuda10 usage.
conda env create -f ymls/cpu.yml
We use Molecular Sets (https://github.com/molecularsets/moses) to train our model : After installing the moses python library, the data can be reached by running
python data_processing/download_moses.py
To train a graph2selfies model, selfies need to be precomputed for the train set by running To compute selfies for another dataset stored in csv, the molecules should be in a column entitled 'smiles', run :
python data_processing/get_selfies.py -i [path_to_my_csv_dataset]
To train the model run
python train.py --train [my_dataset.csv] --n [your_model_name]
The csv must contain columns entitled 'smiles' and 'selfies'
To compute embeddings for molecules in csv file:
python embed_mols.py -i [path_to_csv] --name [your_model_name] -v [smiles]/[selfies]
The column containing the smiles/selfies should be labeled 'smiles'.
To generate samples from a trained model, run :
python generate/sample_prior.py -N [number_of_samples] --name [name_of_the_model] -b [use beam search decoding]
To compute the Moses benchmark metrics for the samples (recommended 30k samples), run
python eval/moses_metrics.py -i [path_to_txt_with_samples]
- The Bayesian Optimization baseline :
/optim - Our implementation of Conditioning By Adaptive Sampling (https://arxiv.org/abs/1901.10060):
/cbas