A molecular generation model based on deep learning algorithm. The generation task is implemented in the TensorFlow framework, allowing the user to run the model to generate a focused library of drug-like molecules.
Refer to requirement.txt
- Install python 3.7 in Linux or Windows.
- If you want to run on a GPU, you will need to install CUDA and cuDNN, please refer to their websites for corresponding versions.
- Add the installation path and run the following command to install all the environment libraries in one step.
pip install -r requirement.txt
You need to open main.py, run load_weights to read the pre-trained weights and get the generated molecules. Or provide training set molecules into coding for model training.
Folder datasets contains pre-training or transfer learning molecules, folder generate contains the molecules generated after transfer learning, folder gen_data is based on the generation model or machine learning model screening molecules.