I wonder if `prune_tree_from_current_node` is a more consistent name taking into account the naming you used for `apply_multiclass` and `apply_multiclass_from_current_node`. I understand you are using a similar design here.