diff options
| author | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2019-12-04 13:09:12 +0000 |
|---|---|---|
| committer | Gertjan van den Burg <gertjanvandenburg@gmail.com> | 2019-12-04 13:09:12 +0000 |
| commit | 51fd39469920122f21fcb89f44f7977b2ac323cb (patch) | |
| tree | 334b1d970c17ada00384d0fa7a7184485860c0b5 | |
| parent | Deal with changing behavior of check_is_fitted (diff) | |
| download | pygensvm-51fd39469920122f21fcb89f44f7977b2ac323cb.tar.gz pygensvm-51fd39469920122f21fcb89f44f7977b2ac323cb.zip | |
Formatting
| -rw-r--r-- | gensvm/sklearn_util.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/gensvm/sklearn_util.py b/gensvm/sklearn_util.py index e23921b..eb8ceb6 100644 --- a/gensvm/sklearn_util.py +++ b/gensvm/sklearn_util.py @@ -89,7 +89,9 @@ 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 @@ -160,7 +162,9 @@ 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 |