fixed reverse sorting issue in label_binarize()

mheilman · amueller · commit 28546c8b16c4 · 2015-04-06T11:36:14.000-04:00
diff --git a/sklearn/preprocessing/label.py b/sklearn/preprocessing/label.py
@@ -559,7 +559,7 @@ def label_binarize(y, classes, neg_label=0, pos_label=1,
 
     # preserve label ordering
     if np.any(classes != sorted_class):
-        indices = np.argsort(classes)
+        indices = np.searchsorted(sorted_class, classes)
         Y = Y[:, indices]
 
     if y_type == "binary":
