These are preliminary reports that have not been peer-reviewed. They should not be regarded as conclusive, guide clinical practice/health-related behavior, or be reported in news media as established information. For more information, please see our FAQs.
Lovric2020_solubility.pdf (739.17 kB)

Machine Learning in Prediction of Intrinsic Aqueous Solubility of Drug-like Compounds: Generalization, Complexity or Predictive Ability?

submitted on 31.07.2020, 16:34 and posted on 04.08.2020, 09:47 by Mario Lovric, Kristina Pavlović, Petar Žuvela, Adrian Spataru, Bono Lučić, Roman Kern, Richard Ming Wah Wong
Here, we present a collection of publicly available
intrinsic aqueous solubility data of 829 drug-like
compounds. Four different machine learning algorithms
(random forest, light GBM, partial least squares and
LASSO) coupled with multi-stage permutation
importance for feature selection and Bayesian hyperparameter optimization were employed for the
prediction of solubility based on chemical structural
information. Our results have shown that LASSO
yielded the best predictive ability on an external test set
with and RMSE(test) of 0.70 log points and 105 features
in the model. Taking into account the number of
descriptors as well, an RF model achieved the best
balance between complexity and predictive ability with
an RMSE(test) of 0.72 with only 17 features. 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
involved and test set performance

The data related to this paper can be downloaded from 10.5281/zenodo.3968754


Email Address of Submitting Author





ORCID For Submitting Author


Declaration of Conflict of Interest

The authors declare no conflict of interest.