We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8157a3e commit d508c22Copy full SHA for d508c22
sklearn/multiclass.py
@@ -302,7 +302,7 @@ def predict(self, X):
302
pred = _predict_binary(e, X)
303
np.maximum(maxima, pred, out=maxima)
304
argmaxima[maxima == pred] = i
305
- return self.classes_[np.array(argmaxima.T)]
+ return self.classes_[argmaxima]
306
else:
307
indices = array.array('i')
308
indptr = array.array('i', [0])
0 commit comments