Skip to content

Commit d508c22

Browse files
MNT Minor clean up in OneVsRestClassifier (scikit-learn#13675)
1 parent 8157a3e commit d508c22

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sklearn/multiclass.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -302,7 +302,7 @@ def predict(self, X):
302302
pred = _predict_binary(e, X)
303303
np.maximum(maxima, pred, out=maxima)
304304
argmaxima[maxima == pred] = i
305-
return self.classes_[np.array(argmaxima.T)]
305+
return self.classes_[argmaxima]
306306
else:
307307
indices = array.array('i')
308308
indptr = array.array('i', [0])

0 commit comments

Comments
 (0)