Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/molML/MoleculeACE
A tool for evaluating the predictive performance on activity cliff compounds of machine learning models
https://github.com/molML/MoleculeACE
Last synced: about 2 months ago
JSON representation
A tool for evaluating the predictive performance on activity cliff compounds of machine learning models
- Host: GitHub
- URL: https://github.com/molML/MoleculeACE
- Owner: molML
- License: mit
- Created: 2022-03-26T08:29:57.000Z (over 2 years ago)
- Default Branch: main
- Last Pushed: 2024-04-10T09:26:00.000Z (5 months ago)
- Last Synced: 2024-07-31T04:57:18.984Z (about 2 months ago)
- Language: Python
- Size: 24.4 MB
- Stars: 156
- Watchers: 4
- Forks: 19
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
- awesome-chemistry-datasets - MoleculeACE
README
[](https://doi.org/10.1021/acs.jcim.2c01073)
Molecule Activity Cliff Estimation (**MoleculeACE**) is a tool for evaluating the predictive performance on activity cliff compounds of machine learning models.
MoleculeACE can be used to:
1) Analyze and compare the performance on activity cliffs of machine learning methods typically employed in
QSAR.
2) Identify best practices to enhance a model’s predictivity in the presence of activity cliffs.
3) Design guidelines to consider when developing novel QSAR approaches.
Update:
**Upon request, we added an extra column to the datasets containing pEC50 and pKi values calculated from Molar concentrations alongside the original training labels used in the study that used log-transformed nM concentrations. Model errors will be the same when trained with either log transformed nM or log transformed M values (except for random processes), since labels are simple shiften by 9.**
:book: Table of Contents
Table of Contents
Benchmark study
In a benchmark study we collected and curated bioactivity data on 30 macromolecular targets, which were used to evaluate
the performance of many machine learning algorithms on activity cliffs. We used classical machine learning methods
combined with common molecular descriptors and neural networks based on unstructured molecular data like molecular
graphs or SMILES strings.
**Activity cliffs are molecules with small differences in structure but large differences in potency.** Activity cliffs
play an important role in drug discovery, but the bioactivity of activity cliff compounds are notoriously difficult to
predict.
*Example of an activity cliff on the Dopamine D3 receptor, D3R*
Tool
Any regression model can be evaluated on activity cliff performance using MoleculeACE on third party data or the 30
included molecular bioactivity data sets. All 24 machine learning strategies covered in our benchmark study can be used
out of the box.
Prerequisites
MoleculeACE currently supports Python 3.8. Some required deep learning packages are not included in the pip install.
- [Tensorflow](https://www.tensorflow.org/) (2.9.0)
- [PyTorch](https://pytorch.org/) (1.11.0)
- [PyTorch Geometric](https://pytorch-geometric.readthedocs.io/en/latest/) (2.0.4)
- [Transformers](https://huggingface.co/docs/transformers/installation) (4.20.1)
Installation
Pip installation
MoleculeACE can be installed as
```pip install MoleculeACE```
Manual installation
```git clone https://github.com/molML/MoleculeACE.git```
```
pip install rdkit-pypi pandas numpy pandas chembl_webresource_client scikit-learn matplotlib tqdm python-Levenshtein
```
Getting started
Train an out-of-the-box model on one of the many included datasets
```python
from MoleculeACE import MPNN, Data, Descriptors, calc_rmse, calc_cliff_rmse, get_benchmark_config
dataset = 'CHEMBL2034_Ki'
descriptor = Descriptors.GRAPH
algorithm = MPNN
# Load data
data = Data(dataset)
# Get the already optimized hyperparameters
hyperparameters = get_benchmark_config(dataset, algorithm, descriptor)
# Featurize SMILES strings with a specific method
data(descriptor)
# Train and a model
model = algorithm(**hyperparameters)
model.train(data.x_train, data.y_train)
y_hat = model.predict(data.x_test)
# Evaluate your model on activity cliff compounds
rmse = calc_rmse(data.y_test, y_hat)
rmse_cliff = calc_cliff_rmse(y_test_pred=y_hat, y_test=data.y_test, cliff_mols_test=data.cliff_mols_test)
print(f"rmse: {rmse}")
print(f"rmse_cliff: {rmse_cliff}")
```
Evaluate the performance of your own model
```python
from MoleculeACE import calc_rmse, calc_cliff_rmse
# Train your own model
model = ...
y_hat = model.predict(...)
# Evaluate your model on activity cliff compounds
rmse = calc_rmse(y_test, y_hat)
# You need to provide both the predicted and true values of the test set + train labels + the train and test molecules
# Activity cliffs are calculated on the fly
rmse_cliff = calc_cliff_rmse(y_test_pred=y_hat, y_test=y_test, smiles_test=smiles_test, y_train=y_train,
smiles_train=smiles_train, in_log10=True, similarity=0.9, potency_fold=10)
print(f"rmse: {rmse}")
print(f"rmse_cliff: {rmse_cliff}")
```
How to cite
Exposing the Limitations of Molecular Machine Learning with Activity Cliffs. Derek van Tilborg, Alisa Alenicheva, and Francesca Grisoni.
Journal of Chemical Information and Modeling, 2022, 62 (23), 5938-5951.
DOI: 10.1021/acs.jcim.2c01073
License
MoleculeACE is under MIT license. For use of specific models, please refer to the model licenses found in the original
packages.