File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 2727 PatchingConditionsChain )
2828import logging
2929
30- from sklearn .tree import DecisionTreeClassifier
30+ from sklearn .tree import DecisionTreeClassifier , DecisionTreeRegressor
3131from sklearn .tree ._tree import Tree
3232from sklearn .ensemble import RandomForestClassifier as RandomForestClassifier_original
3333from sklearn .ensemble import RandomForestRegressor as RandomForestRegressor_original
@@ -1052,7 +1052,7 @@ def _estimators_(self):
10521052 }
10531053 if not sklearn_check_version ('1.0' ):
10541054 params ['min_impurity_split' ] = self .min_impurity_split
1055- est = DecisionTreeClassifier (** params )
1055+ est = DecisionTreeRegressor (** params )
10561056
10571057 # we need to set est.tree_ field with Trees constructed from Intel(R)
10581058 # oneAPI Data Analytics Library solution
You can’t perform that action at this time.
0 commit comments