SchNetPack Hyperparameter Optimization for a More Reliable Top Docking Scores Prediction

Ján Matúška,Lukas Bucinsky,Marián Gall,Michal Pitoňák,Marek Štekláč
DOI: https://doi.org/10.1021/acs.jpcb.4c00296
IF: 3.466
2024-05-12
The Journal of Physical Chemistry B
Abstract:Options to improve the extrapolation power of the neural network designed using the SchNetPack package with respect to top docking scores prediction are presented. It is shown that hyperparameter tuning of the atomistic model representation (in the schnetpack.representation) improves the prediction of the top scoring compounds, which have characteristically a low incidence in randomized data sets for training of machine learning models. The prediction robustness is evaluated according to the...
chemistry, physical
What problem does this paper attempt to address?