Skip to content

Commit 77c963d

Browse files
committed
Merge pull request #5230 from jmschrei/_tree_split
[MRG+1] split tree module into several packages
2 parents 7fccf45 + 5978c0b commit 77c963d

19 files changed

+48573
-35262
lines changed

sklearn/ensemble/_gradient_boosting.c

+829-652
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

sklearn/ensemble/_gradient_boosting.pyx

+3-1
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,9 @@ import numpy as np
1212
cimport numpy as np
1313
np.import_array()
1414

15-
from sklearn.tree._tree cimport Tree, Node
15+
from sklearn.tree._tree cimport Node
16+
from sklearn.tree._tree cimport Tree
17+
1618

1719
ctypedef np.int32_t int32
1820
ctypedef np.float64_t float64

sklearn/ensemble/gradient_boosting.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,8 @@
4545

4646
from ..tree.tree import DecisionTreeRegressor
4747
from ..tree._tree import DTYPE, TREE_LEAF
48-
from ..tree._tree import PresortBestSplitter
49-
from ..tree._tree import FriedmanMSE
48+
from ..tree._splitter import PresortBestSplitter
49+
from ..tree._criterion import FriedmanMSE
5050

5151
from ._gradient_boosting import predict_stages
5252
from ._gradient_boosting import predict_stage

0 commit comments

Comments
 (0)