rbartpackages.missBART.TreePriorParams

class rbartpackages.missBART.TreePriorParams[source]

Tree-prior parameters, the keys of R’s tree_list.

The prior probability of splitting a node at depth d is prior_alpha * (1 + d) ** -prior_beta. Entries left out fall back to tree_list’s own defaults, noted below.

prior_alpha: float

Base of the node-splitting probability, in (0, 1). Default 0.95.

prior_beta: float

Power of the node-splitting probability, nonnegative. Default 2.

min_node: int

Minimum number of observations per leaf, raised to p + 1 internally. Default 1.

max_attempt: int

Number of attempts to propose a valid split before giving up. Default 1.