Published March 30, 2021 | Version v1.0

astuke/HyperTune: Optimization of machine learning hyperparameters in chemical physics

Authors/Creators

  • 1. Aalto University

Description

Python code for the optimization of hyperparameters in machine learning (ML), applied to a problem in chemical physics. The ML model at hand is based on kernel ridge regression (KRR) and predicts molecular orbital energies. The hyperparameters in this setup stem from two sources: the KRR method itself and the descriptors for the atomic structure of molecules, resulting in the simultaneous optimization of up two 6 hyperparameters. This repository includes code for three different optimization methods: Bayesian optimization (BO), grid search and random search.

Files

astuke/HyperTune-v1.0.zip

Files (363.1 kB)

Name Size Download all
md5:9f3df37f0cf0b5080cab09e36ae9f455
363.1 kB Preview Download

Additional details

Related works