diff options
| author | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2019-03-06 12:24:41 -0500 |
|---|---|---|
| committer | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2019-03-06 12:24:41 -0500 |
| commit | 4787e9c6d5a5625948240b055a98f15249622a3f (patch) | |
| tree | a0219246f3a6e758357e3d5b397ed78c04e14cb3 /gensvm | |
| parent | Add support for interrupted grid search (diff) | |
| download | pygensvm-4787e9c6d5a5625948240b055a98f15249622a3f.tar.gz pygensvm-4787e9c6d5a5625948240b055a98f15249622a3f.zip | |
Formatting
Diffstat (limited to 'gensvm')
| -rw-r--r-- | gensvm/gridsearch.py | 2 | ||||
| -rw-r--r-- | gensvm/sklearn_util.py | 8 |
2 files changed, 2 insertions, 8 deletions
diff --git a/gensvm/gridsearch.py b/gensvm/gridsearch.py index d62ab1d..3fdfa9a 100644 --- a/gensvm/gridsearch.py +++ b/gensvm/gridsearch.py @@ -573,8 +573,6 @@ class GenSVMGridSearchCV(BaseEstimator, MetaEstimatorMixin): self.iid, ) - self.cv_results_ = results - # For multi-metric evaluation, store the best_index_, best_params_ and # best_score_ iff refit is one of the scorer names # In single metric evaluation, refit_metric is "score" diff --git a/gensvm/sklearn_util.py b/gensvm/sklearn_util.py index 05d9618..182f257 100644 --- a/gensvm/sklearn_util.py +++ b/gensvm/sklearn_util.py @@ -89,9 +89,7 @@ def _skl_format_cv_results( score_time, ) = zip(*out) else: - (test_score_dicts, test_sample_counts, fit_time, score_time) = zip( - *out - ) + (test_score_dicts, test_sample_counts, fit_time, score_time) = zip(*out) # test_score_dicts and train_score dicts are lists of dictionaries and # we make them into dict of lists @@ -162,9 +160,7 @@ def _skl_format_cv_results( ) if return_train_score: _store( - "train_%s" % scorer_name, - train_scores[scorer_name], - splits=True, + "train_%s" % scorer_name, train_scores[scorer_name], splits=True ) return results |