bartz.grove.forest_mean_leaves

bartz.grove.forest_mean_leaves(split_tree)

Return the average number of leaves per tree in a set of trees.

Parameters:

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

Returns:

Float32[Array, ''] – The mean number of leaves across the trees.