transition_ratio_grow: Transition ratio for grow.
In brunaw/hebart: Hierachical Embedded Bayesian Additive Regression Trees

transition_ratio_grow

R Documentation

Transition ratio for grow.

Description

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

Usage

transition_ratio_grow(tree, current_node, p_vars, current_selec_var, p_grow, i)

Arguments

`tree`	The current tree
`current_node`	The current node
`p_vars`	The number of predictors still available
`current_selec_var`	The variable selected for the split
`p_grow`	The probability of growing the tree
`i`	The current iteration

Details

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