Skip to content

Commit 5978c0b

Browse files
jmschreijmschrei
committed
ENH split _tree.pyx into several files
1 parent 7fccf45 commit 5978c0b

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)