bartz.grove.forest_depth_distr

bartz.grove.forest_depth_distr(split_tree)

Histogram the depths of a set of trees.

Parameters:

split_tree (UInt[Array, '*batch_shape num_trees half_tree_size']) – The cutpoints of the decision rules of the trees.

Returns:

Int32[Array, '*batch_shape d'] – An integer vector where the i-th element counts how many trees have depth i.