Changing dtypes for new_labels

mjbommar · mjbommar · commit bb8d9a647252 · 2014-06-08T14:04:07.000-04:00
diff --git a/sklearn/preprocessing/label.py b/sklearn/preprocessing/label.py
@@ -167,7 +167,7 @@ def transform(self, y):
             elif self.new_labels == "nan":
                 # Create copy of array and return
                 y_array = np.array(y)
-                z = np.zeros(y_array.shape)
+                z = np.zeros(y_array.shape, dtype=float)
 
                 # Find entries with new labels
                 missing_mask = np.in1d(y, diff)
@@ -180,7 +180,7 @@ def transform(self, y):
             elif self.new_labels == "label":
                 # Create copy of array and return
                 y_array = np.array(y)
-                z = np.zeros(y_array.shape)
+                z = np.zeros(y_array.shape, dtype=int)
 
                 # Find entries with new labels
                 missing_mask = np.in1d(y, diff)
