original version presented in S. Riniker, G. Landrum, J. Cheminf., 5, 26 (2013), DOI: 10.1186/1758-2946-5-26, URL: http://www.jcheminf.com/content/5/1/26
extended version presented in S. Riniker, N. Fechner, G. Landrum, J. Chem. Inf. Model., 53, 2829, (2013), DOI: 10.1021/ci400466r, URL: http://pubs.acs.org/doi/abs/10.1021/ci400466r
Our released models are listed as following. You can import these models by using the smiles-featurizers
package or using HuggingFace's Transformers.
Model | Type | AUROC | BEDROC |
---|---|---|---|
shahrukhx01/smole-bert | Bert |
0.615 |
0.225 |
shahrukhx01/smole-bert-mtr | Bert |
0.621 |
0.262 |
shahrukhx01/smole-bart | Bart |
0.660 |
0.263 |
shahrukhx01/muv2x-simcse-smole-bart | Simcse |
0.697 |
0.270 |
shahrukhx01/siamese-smole-bert-muv-1x | SentenceTransformer |
0.673 |
0.274 |
The virtual-screening process implemented by the benchmarking platform is divided into three steps:
cd scoring/data_sets_1/
mkdir <name-of-fingerprints-directory>
python calculate_scored_lists_transformers.py -n 5 -f config.txt -s EmbedCosine -m shahrukhx01/smole-bert -t bert -o <name-of-fingerprints-directory>
cd ../validation/data_sets_1/
cp -r ../scoring/data_sets_1/<name-of-fingerprints-directory> .
mkdir <name-of-validation-results-directory>
python calculate_validation_methods.py -m methods.txt -i fingerprints -o <name-of-validation-results-directory> /
cd ../analysis/data_sets_1/
cp -r ../validation/data_sets_1/<name-of-validation-results-directory> .
mkdir <name-of-analysis-results-directory>
python run_analysis.py -i validation_results -o <name-of-analysis-results-directory>/
The three steps are run separately and read in the output of the previous step. In the scoring step, the data from the directories compounds and query_lists is read in.
The directory compounds contains lists of compounds for 118 targets from three public data sources: MUV, DUD and ChEMBL. The compound lists contain the external ID, the internal ID and the SMILES of each compound.
There are three subsets of targets available:
subset I: 88 targets from MUV, DUD & ChEMBL described in J. Cheminf., 5, 26 (2013)
subset I filtered: 69 targets from MUV, DUD & ChEMBL filtered for difficulty described in JCIM (2013), online
subset II: 37 targets from ChEMBL designed for a second VS use case described in JCIM (2013), online
The directory query_lists contains training lists for each target with the indices of randomly selected active and inactive molecules. Training lists with 5, 10 or 20 active molecules are available. The number of training decoys is 20 % of the decoys for subsets I and 10 % for subset II.
The scripts are written in Python and use the open-source cheminformatics library RDKit (www.rdkit.org) and machine-learning library scikit-learn (www.scikit-learn.org).
Running a script with the option [--help] gives a description of the required and optional input parameters of the script.