PPPLDeepLearning / plasma-python

PPPL deep learning disruption prediction package
http://tigress-web.princeton.edu/~alexeys/docs-web/html/
79 stars 43 forks source link

Random hyperpar search #5

Closed ASvyatkovskiy closed 7 years ago

ASvyatkovskiy commented 7 years ago

This PR introduces the hyperparameter search. The current version closely follows the random parameter optimization as implemented in hyperopt package.

This PR is multiprocessing version (non-distributed).

The key optimization logic is in the HyperRunner class and an additional ModelBuilder.hyper_build_model method. The latter should be modified and possibly included in the ModelBuilder.build_model method using conf instead of space dictionary.

HyperRunner class introduces: keras_fmin_fnct get_space and frnn_minimize

The distributed version and API changes to improve user experience will be added in the following PRs