Skip to content

Commit 68c9be4

Browse files
md-shafiul-alamnapetrov
authored andcommitted
Replace DTClassifier with DTRegressor in RFRegressor (#1052)
1 parent 54ac92a commit 68c9be4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

daal4py/sklearn/ensemble/_forest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
PatchingConditionsChain)
2828
import logging
2929

30-
from sklearn.tree import DecisionTreeClassifier
30+
from sklearn.tree import DecisionTreeClassifier, DecisionTreeRegressor
3131
from sklearn.tree._tree import Tree
3232
from sklearn.ensemble import RandomForestClassifier as RandomForestClassifier_original
3333
from 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

0 commit comments

Comments
 (0)