transition_ratio_prune: Transition ratio for prune
In brunaw/hebart: Hierachical Embedded Bayesian Additive Regression Trees

transition_ratio_prune

R Documentation

Transition ratio for prune

Description

Transition probability of going to the candidate tree, given that the action step was a prune

Usage

transition_ratio_prune(
  old_tree,
  tree,
  current_node,
  p_split,
  var_in_prune,
  i,
  p_grow
)

Arguments

`old_tree`	The previous tree
`tree`	The current tree
`current_node`	The current node
`p_split`	Probability of splitting a variable
`var_in_prune`	The variable that was split in the node chosen to be pruned.
`i`	The current iteration number
`p_grow`	The grow probability

Details

When transitioning to a prune, we need the probabilities of: 1. Pruning the tree 2. Selecting node to prune When transitioning from a prune back to the split, we need the probabilities of: 1. Growing the tree 2. Growing from the specific pruned node, that has to consider the available predictors and available values for that grow