Skip to main content

ChemSpaceAL Python package: an efficient active learning methodology applied to protein-specific molecular generation

Project description

ChemSpaceAL: An Efficient Active Learning Methodology Applied to Protein- Specific Molecular Generation

Checked with mypy Code style: black codecov License: MIT image Open In Colab

A description of the active learning methodology

Abstract

The incredible capabilities of generative artificial intelligence models have inevitably led to their application in the domain of drug discovery. Within this domain, the vastness of chemical space motivates the development of more efficient methods for identifying regions with molecules that exhibit desired characteristics. In this work, we present a computationally efficient active learning methodology that requires evaluation of only a subset of the generated data in the constructed sample space to successfully align a generative model with respect to a specified objective. We demonstrate the applicability of this methodology to targeted molecular generation by fine-tuning a GPT-based molecular generator toward a protein with FDA-approved small-molecule inhibitors, c-Abl kinase. Remarkably, the model learns to generate molecules similar to the inhibitors without prior knowledge of their existence, and even reproduces two of them exactly. We also show that the methodology is effective for a protein without any commercially available small-molecule inhibitors, the HNH domain of the CRISPR-associated protein 9 (Cas9) enzyme. We believe that the inherent generality of this method ensures that it will remain applicable as the exciting field of in silico molecular generation evolves. To facilitate implementation and reproducibility, we have made all of our software available through the open-source ChemSpaceAL Python package.

Preprint

Associated preprint can be found on arXiv. Note, a second version of the preprint has been posted on Dec 4, 2023.

Installation

in order to install the ChemSpaceAL package, simply run:

pip install ChemSpaceAL

You could also open ChemSpaceAL.ipynb in Google Colab to see an example of how to use a package.

Contact

Please feel free to reach out to us through either of the following emails if you have any questions or need any additional files:

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ChemSpaceAL-2.0.1.tar.gz (28.2 kB view hashes)

Uploaded Source

Built Distribution

ChemSpaceAL-2.0.1-py3-none-any.whl (32.1 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page