File tree Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Expand file tree Collapse file tree 1 file changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -982,7 +982,7 @@ cdef class BestFirstTreeBuilder(TreeBuilder):
982
982
cdef class BaseTree:
983
983
""" Base class for Cython tree models.
984
984
985
- Downstream classes must implement
985
+ Downstream classes must implement methods to actually traverse the tree.
986
986
"""
987
987
cdef int _resize(
988
988
self ,
@@ -1712,6 +1712,10 @@ cdef class Tree(BaseTree):
1712
1712
weighted_n_node_samples : array of double, shape [node_count]
1713
1713
weighted_n_node_samples[i] holds the weighted number of training samples
1714
1714
reaching node i.
1715
+
1716
+ leaf_node_samples : dict of node id to numpy array of shapes (n_samples_node, n_features)
1717
+ A dictionary mapping leaf nodes to the samples of data that are used
1718
+ to fit the prediction at each leaf.
1715
1719
"""
1716
1720
# Wrap for outside world.
1717
1721
# WARNING: these reference the current `nodes` and `value` buffers, which
You can’t perform that action at this time.
0 commit comments