simplified code to find best hyperparams in experiment