forest uses DOUBLE for y

amueller · Jul 21, 2012 · ea8c899 · ea8c899
1 parent 2cce14c
commit ea8c899
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sklearn/ensemble/forest.py b/sklearn/ensemble/forest.py
@@ -44,7 +44,7 @@ class calls the ``fit`` method of each sub-estimator on random samples
 from ..feature_selection.selector_mixin import SelectorMixin
 from ..tree import DecisionTreeClassifier, DecisionTreeRegressor, \
                    ExtraTreeClassifier, ExtraTreeRegressor
-from ..tree._tree import DTYPE
+from ..tree._tree import DTYPE, DOUBLE
 from ..utils import array2d, check_random_state
 from ..metrics import r2_score
 
@@ -269,7 +269,7 @@ def fit(self, X, y):
                 y[:, k] = np.searchsorted(unique, y[:, k])
 
         if getattr(y, "dtype", None) != DTYPE or not y.flags.contiguous:
-            y = np.ascontiguousarray(y, dtype=DTYPE)
+            y = np.ascontiguousarray(y, dtype=DOUBLE)
 
         # Assign chunk of trees to jobs
         n_jobs, n_trees, _ = _partition_trees(self)