Skip to content

Commit 8095921

Browse files
committed
Fixed
Signed-off-by: Adam Li <adam2392@gmail.com>
1 parent 5b074dd commit 8095921

File tree

2 files changed

+21
-13
lines changed

2 files changed

+21
-13
lines changed

sklearn/ensemble/_forest.py

Lines changed: 16 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -36,10 +36,10 @@ class calls the ``fit`` method of each sub-estimator on random samples
3636
# SPDX-License-Identifier: BSD-3-Clause
3737

3838

39-
from time import time
4039
import threading
4140
from abc import ABCMeta, abstractmethod
4241
from numbers import Integral, Real
42+
from time import time
4343
from warnings import catch_warnings, simplefilter, warn
4444

4545
import numpy as np
@@ -54,22 +54,20 @@ class calls the ``fit`` method of each sub-estimator on random samples
5454
_fit_context,
5555
is_classifier,
5656
)
57+
from sklearn.ensemble._base import BaseEnsemble, _partition_estimators
58+
from sklearn.ensemble._hist_gradient_boosting.binning import _BinMapper
5759
from sklearn.exceptions import DataConversionWarning
5860
from sklearn.metrics import accuracy_score, r2_score
5961
from sklearn.preprocessing import OneHotEncoder
60-
from ..tree import (
61-
BaseDecisionTree,
62-
DecisionTreeClassifier,
63-
DecisionTreeRegressor,
64-
ExtraTreeClassifier,
65-
ExtraTreeRegressor,
66-
)
67-
from ..tree._tree import DOUBLE, DTYPE
6862
from sklearn.utils import check_random_state, compute_sample_weight
6963
from sklearn.utils._openmp_helpers import _openmp_effective_n_threads
7064
from sklearn.utils._param_validation import Interval, RealNotInt, StrOptions
7165
from sklearn.utils._tags import get_tags
72-
from sklearn.utils.multiclass import check_classification_targets, type_of_target
66+
from sklearn.utils.multiclass import (
67+
_check_partial_fit_first_call,
68+
check_classification_targets,
69+
type_of_target,
70+
)
7371
from sklearn.utils.parallel import Parallel, delayed
7472
from sklearn.utils.validation import (
7573
_check_feature_names_in,
@@ -78,9 +76,15 @@ class calls the ``fit`` method of each sub-estimator on random samples
7876
check_is_fitted,
7977
validate_data,
8078
)
81-
from sklearn.ensemble._hist_gradient_boosting.binning import _BinMapper
82-
from ._base import BaseEnsemble, _partition_estimators
8379

80+
from ..tree import (
81+
BaseDecisionTree,
82+
DecisionTreeClassifier,
83+
DecisionTreeRegressor,
84+
ExtraTreeClassifier,
85+
ExtraTreeRegressor,
86+
)
87+
from ..tree._tree import DOUBLE, DTYPE
8488

8589
__all__ = [
8690
"RandomForestClassifier",

sklearn/tree/_classes.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,10 @@
2626
)
2727
from sklearn.utils import Bunch, check_random_state, compute_sample_weight
2828
from sklearn.utils._param_validation import Hidden, Interval, RealNotInt, StrOptions
29-
from sklearn.utils.multiclass import check_classification_targets
29+
from sklearn.utils.multiclass import (
30+
_check_partial_fit_first_call,
31+
check_classification_targets,
32+
)
3033
from sklearn.utils.validation import (
3134
_assert_all_finite_element_wise,
3235
_check_n_features,
@@ -35,6 +38,7 @@
3538
check_is_fitted,
3639
validate_data,
3740
)
41+
3842
from . import _criterion, _splitter, _tree
3943
from ._criterion import BaseCriterion
4044
from ._splitter import BaseSplitter

0 commit comments

Comments
 (0)