Fix partial fit

Signed-off-by: Adam Li <adam2392@gmail.com>
neurodata · Sep 9, 2024 · dd58597 · dd58597
1 parent 4551602
commit dd58597
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 23 deletions.
diff --git a/sklearn/ensemble/_forest.py b/sklearn/ensemble/_forest.py
@@ -1254,15 +1254,7 @@ def partial_fit(self, X, y, sample_weight=None, classes=None):
         self : object
             Returns the instance itself.
         """
-        X, y = validate_data(
-            self,
-            X,
-            y,
-            multi_output=True,
-            accept_sparse="csc",
-            dtype=DTYPE,
-            ensure_all_finite=False,
-        )
+        self._validate_params()
 
         # validate input parameters
         first_call = _check_partial_fit_first_call(self, classes=classes)

diff --git a/sklearn/tree/_classes.py b/sklearn/tree/_classes.py
@@ -1375,15 +1375,7 @@ def partial_fit(self, X, y, sample_weight=None, check_input=True, classes=None):
         self : DecisionTreeClassifier
             Fitted estimator.
         """
-        X, y = validate_data(
-            self,
-            X,
-            y,
-            multi_output=True,
-            accept_sparse="csc",
-            dtype=DTYPE,
-            ensure_all_finite=False,
-        )
+        self._validate_params()
 
         # validate input parameters
         first_call = _check_partial_fit_first_call(self, classes=classes)
@@ -1406,11 +1398,7 @@ def partial_fit(self, X, y, sample_weight=None, check_input=True, classes=None):
             check_X_params = dict(dtype=DTYPE, accept_sparse="csc")
             check_y_params = dict(ensure_2d=False, dtype=None)
             X, y = validate_data(
-                self,
-                X,
-                y,
-                reset=False,
-                validate_separately=(check_X_params, check_y_params),
+                self, X, y, reset=False, validate_separately=(check_X_params, check_y_params)
             )
             if issparse(X):
                 X.sort_indices()