Merge pull request scikit-learn#5796 from max-moroz/dummy_classifier_…

…memory_efficiency [MRG+1] memory issue in DummyClassifier(strategy='most_frequent')
sergiogaiotto · Nov 12, 2015 · ce90ccb · ce90ccb
2 parents 7cb9a23 + ea7c5c3
commit ce90ccb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sklearn/dummy.py b/sklearn/dummy.py
@@ -272,7 +272,7 @@ def predict_proba(self, X):
         P = []
         for k in range(self.n_outputs_):
             if self.strategy == "most_frequent":
-                ind = np.ones(n_samples, dtype=int) * class_prior_[k].argmax()
+                ind = class_prior_[k].argmax()
                 out = np.zeros((n_samples, n_classes_[k]), dtype=np.float64)
                 out[:, ind] = 1.0
             elif self.strategy == "prior":