ChemTS

Molecule Design using Monte Carlo Tree Search with Neural Rollout. ChemTS can design novel molecules with desired properties(such as, HOMO-LUMO gap, energy, logp..). Combining with rDock, ChemTS can design molecules active to target proteins. The ChemTS paper is available at https://arxiv.org/abs/1710.00616 . Also, we introduced the distributed parallel ChemTS that can accerlate molecular discovery. And the distributed parallel ChemTS is available at https://github.com/tsudalab/DP-ChemTS.

Requirements

Python>=2.7
Keras (version 2.0.5) If you installed the newest version of keras, some errors will show up. Please change it back to keras 2.0.5 by pip install keras==2.0.5.
rdkit
rDock

How to use ChemTS?

For usage, please refer the following instructions. Currently, the package hasn't been finished very well... If you want to implement your own simulator, please check add_node_type.py. The full package will be updated later.

Train a RNN model for molecule generation

cd train_RNN
Run python train_RNN.py to train the RNN model. GPU is highly recommended for reducing the training time.

Design materials with desired HOMO-LUMO and internal energy (coming soon)

Design molecules active to target proteins

cd ligand_design
Run python mcts_ligand.py

MCTS for logP optimization

There are two versions of chemts for logP optimization. The search tree of the old version added all possible smiles symbols as children nodes. While the new version chemts only expands children nodes with high probabilities predicted by RNN.

cd mcts_logp_improved_version
Run python mcts_logp.py

License

This package is distributed under the MIT License.

mygmyg / ChemTS