Remove class padding behavior in 1 class case

hamsal · hamsal · commit c99f1b9363ee · 2014-05-29T17:16:43.000-04:00
diff --git a/sklearn/preprocessing/label.py b/sklearn/preprocessing/label.py
@@ -282,11 +282,11 @@ def fit(self, y):
             self.sparse_indicator_matrix_ = issparse(y)
 
         self.classes_ = unique_labels(y)
-        if self.y_type_ == "binary" and len(self.classes_) == 1:
-            if unique_labels(y)[0] == self.pos_label:
-                self.classes_ = np.insert(self.classes_, [self.neg_label], 0)
-            else:
-                self.classes_ = np.insert(self.classes_, [self.pos_label], 0)
+        # if self.y_type_ == "binary" and len(self.classes_) == 1:
+        #     if unique_labels(y)[0] == self.pos_label:
+        #         self.classes_ = np.insert(self.classes_, [self.neg_label], 0)
+        #     else:
+        #         self.classes_ = np.insert(self.classes_, [self.pos_label], 0)
         return self
 
     def transform(self, y):
