From 4d37cf58626bb7fc7d28a5c323faed11ba3b6ee4 Mon Sep 17 00:00:00 2001
From: Joel Nothman <joel.nothman@gmail.com>
Date: Wed, 5 Jul 2017 17:44:06 +1000
Subject: [PATCH] COSMIT use zip instead of next in generator expression
(#9281)
---
sklearn/multiclass.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/sklearn/multiclass.py b/sklearn/multiclass.py
index 97c874d4dd..3ca3b1ad42 100644
--- a/sklearn/multiclass.py
+++ b/sklearn/multiclass.py
@@ -267,9 +267,8 @@ class OneVsRestClassifier(BaseEstimator, ClassifierMixin, MetaEstimatorMixin):
columns = (col.toarray().ravel() for col in Y.T)
self.estimators_ = Parallel(n_jobs=self.n_jobs)(
- delayed(_partial_fit_binary)(self.estimators_[i], X,
- next(columns))
- for i in range(self.n_classes_))
+ delayed(_partial_fit_binary)(estimator, X, column)
+ for estimator, column in izip(self.estimators_, columns))
return self
--
GitLab