Skip to content

Commit

Permalink
Update docstring
Browse files Browse the repository at this point in the history
Signed-off-by: Adam Li <adam2392@gmail.com>
  • Loading branch information
adam2392 committed Sep 8, 2023
1 parent 4627339 commit 62f0c60
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion sklearn/tree/_tree.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -982,7 +982,7 @@ cdef class BestFirstTreeBuilder(TreeBuilder):
cdef class BaseTree:
"""Base class for Cython tree models.
Downstream classes must implement
Downstream classes must implement methods to actually traverse the tree.
"""
cdef int _resize(
self,
Expand Down Expand Up @@ -1712,6 +1712,10 @@ cdef class Tree(BaseTree):
weighted_n_node_samples : array of double, shape [node_count]
weighted_n_node_samples[i] holds the weighted number of training samples
reaching node i.
leaf_node_samples : dict of node id to numpy array of shapes (n_samples_node, n_features)
A dictionary mapping leaf nodes to the samples of data that are used
to fit the prediction at each leaf.
"""
# Wrap for outside world.
# WARNING: these reference the current `nodes` and `value` buffers, which
Expand Down

0 comments on commit 62f0c60

Please sign in to comment.