diff options
| author | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2019-01-15 16:57:07 +0000 |
|---|---|---|
| committer | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2019-01-15 16:57:07 +0000 |
| commit | 54506be3f8e89f0a3f52242863dc3cbb4186bd31 (patch) | |
| tree | 765af9be62004bb9baa8f1d1afc2a4213eb2f800 /gensvm/gridsearch.py | |
| parent | Update submodule (diff) | |
| download | pygensvm-54506be3f8e89f0a3f52242863dc3cbb4186bd31.tar.gz pygensvm-54506be3f8e89f0a3f52242863dc3cbb4186bd31.zip | |
Update grid search helper to add verbosity
Diffstat (limited to 'gensvm/gridsearch.py')
| -rw-r--r-- | gensvm/gridsearch.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/gensvm/gridsearch.py b/gensvm/gridsearch.py index bf4b9ce..fbb9168 100644 --- a/gensvm/gridsearch.py +++ b/gensvm/gridsearch.py @@ -245,7 +245,13 @@ def _fit_grid_gensvm( fold_idx += 1 results_ = wrapper.grid_wrap( - X, y, candidate_params, int(store_predictions), cv_idx, int(n_folds) + X, + y, + candidate_params, + int(store_predictions), + cv_idx, + int(n_folds), + int(verbose), ) cv_results_ = _format_results(results_, cv_idx, y, scorers, iid) |