+ search hyper params: code for grid search on all combinations and call the training func