DOC Note shape of binary binarized output

jnothman · Hamzeh Alsalhi · commit 34872e9c73e5 · 2014-06-24T21:17:05.000-04:00
diff --git a/sklearn/preprocessing/label.py b/sklearn/preprocessing/label.py
@@ -287,6 +287,7 @@ def transform(self, y):
         Returns
         -------
         Y : numpy array of shape [n_samples, n_classes]
+            Shape will be [n_samples, 1] for binary problems.
         """
         self._check_fitted()
 
@@ -393,6 +394,7 @@ def label_binarize(y, classes, multilabel=False, neg_label=0, pos_label=1):
     Returns
     -------
     Y : numpy array of shape [n_samples, n_classes]
+        Shape will be [n_samples, 1] for binary problems.
 
     Examples
     --------
