bartz.grove.is_leaves_parent

bartz.grove.is_leaves_parent(split_tree)

Return a mask indicating the nodes with leaf (and only leaf) children.

Parameters:

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

Returns:

Bool[Array, 'half_tree_size'] – The mask indicating which nodes have leaf children.