hrvatski jezikClear Cookie - decide language by browser settings

Machine learning in prediction of intrinsic aqueous solubility of drug‐like compounds: Generalization, complexity, or predictive ability?

Lovrić, Mario; Pavlović, Kristina; Žuvela, Petar; Spataru, Adrian; Lučić, Bono; Kern, Roman; Wong, Ming Wah (2021) Machine learning in prediction of intrinsic aqueous solubility of drug‐like compounds: Generalization, complexity, or predictive ability?. Journal of chemometrics, 35 (7-8). ISSN 0886-9383

PDF - Published Version - article
Available under License Creative Commons Attribution.

Download (2MB) | Preview


We present a collection of publicly available intrinsic aqueous solubility data of 829 drug‐like compounds. Four different machine learning algorithms (random forests [RF], LightGBM, partial least squares, and least absolute shrinkage and selection operator [LASSO]) coupled with multistage permutation importance for feature selection and Bayesian hyperparameter optimization were used for the prediction of solubility based on chemical structural information. Our results show that LASSO yielded the best predictive ability on an external test set with a root mean square error (RMSE) (test) of 0.70 log points, an R2(test) of 0.80, and 105 features. Taking into account the number of descriptors as well, an RF model achieves the best balance between complexity and predictive ability with an RMSE(test) of 0.72 log points, an R2(test) of 0.78, and with only 17 features. On a more aggressive test set (principal component analysis [PCA]‐based split), better generalization was observed for the RF model. We propose a ranking score for choosing the best model, as test set performance is only one of the factors in creating an applicable model. The ranking score is a weighted combination of generalization, number of features, and test performance. Out of the two best learners, a consensus model was built exhibiting the best predictive ability and generalization with RMSE(test) of 0.67 log points and a R2(test) of 0.81.

Item Type: Article
Uncontrolled Keywords: consensus modeling ; LASSO ; LightGBM ; PCA ; permutation importance ; QSAR ; randomforests
Subjects: NATURAL SCIENCES > Chemistry
NATURAL SCIENCES > Interdisciplinary Natural Sciences
Divisions: NMR Center
Project titleProject leaderProject codeProject type
Napredne metode i tehnologije u znanosti o podatcima i kooperativnim sustavimaLončarić, Sven; Ivan, PetrovićKK.
Depositing User: Bono Lučić
Date Deposited: 10 Jan 2022 12:38
DOI: 10.1002/cem.3349

Actions (login required)

View Item View Item


Downloads per month over past year

Increase Font
Decrease Font
Dyslexic Font