Skip to content

Latest commit

 

History

1 Commit

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 

Repository files navigation

Neural Network Hyper Parameter Search with Bayesian Optimization for MNIST Classification

This example demonstrates the usage of Bayesian optimization (BO) as implemented in scikit-optimize for tuning the hyper parameters of a neural network. As compared to grid search or random search less hyper parameter combinations need to be evaluated to determine the near-optimal parameter configuration.

In each iteration of BO the neural network is trained with a different set of hyper parameters and its validation accuracy is computed. A gaussian process is fitted to the validation accuracy over the space of hyper parameters. An acquistion function which is a function of the fitted gaussian process is sampled for the next suitable hyper parameter combination which is expected to increase validation accuracy.

For a great tutorial on Bayesian information read this article.

About

Neural Network Hyper Parameter Search with Bayesian Optimization for MNIST Classification

Resources

Stars

1 star

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages