Skip to content

Commit dd58597

Browse files
committed
Fix partial fit
Signed-off-by: Adam Li <adam2392@gmail.com>
1 parent 4551602 commit dd58597

File tree

2 files changed

+3
-23
lines changed

2 files changed

+3
-23
lines changed

sklearn/ensemble/_forest.py

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1254,15 +1254,7 @@ def partial_fit(self, X, y, sample_weight=None, classes=None):
12541254
self : object
12551255
Returns the instance itself.
12561256
"""
1257-
X, y = validate_data(
1258-
self,
1259-
X,
1260-
y,
1261-
multi_output=True,
1262-
accept_sparse="csc",
1263-
dtype=DTYPE,
1264-
ensure_all_finite=False,
1265-
)
1257+
self._validate_params()
12661258

12671259
# validate input parameters
12681260
first_call = _check_partial_fit_first_call(self, classes=classes)

sklearn/tree/_classes.py

Lines changed: 2 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1375,15 +1375,7 @@ def partial_fit(self, X, y, sample_weight=None, check_input=True, classes=None):
13751375
self : DecisionTreeClassifier
13761376
Fitted estimator.
13771377
"""
1378-
X, y = validate_data(
1379-
self,
1380-
X,
1381-
y,
1382-
multi_output=True,
1383-
accept_sparse="csc",
1384-
dtype=DTYPE,
1385-
ensure_all_finite=False,
1386-
)
1378+
self._validate_params()
13871379

13881380
# validate input parameters
13891381
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):
14061398
check_X_params = dict(dtype=DTYPE, accept_sparse="csc")
14071399
check_y_params = dict(ensure_2d=False, dtype=None)
14081400
X, y = validate_data(
1409-
self,
1410-
X,
1411-
y,
1412-
reset=False,
1413-
validate_separately=(check_X_params, check_y_params),
1401+
self, X, y, reset=False, validate_separately=(check_X_params, check_y_params)
14141402
)
14151403
if issparse(X):
14161404
X.sort_indices()

0 commit comments

Comments
 (0)