Description Usage Arguments Details Value See Also Examples
View source: R/ML_BARTMachineModel.R
Builds a BART model for regression or classification.
1 2 3 4 5 6 7 8 9 10 11 12 13 |
num_trees |
number of trees to be grown in the sum-of-trees model. |
num_burn |
number of MCMC samples to be discarded as "burn-in". |
num_iter |
number of MCMC samples to draw from the posterior distribution. |
alpha, beta |
base and power hyperparameters in tree prior for whether a node is nonterminal or not. |
k |
regression prior probability that E(Y|X) is contained in the interval (y_{min}, y_{max}), based on a normal distribution. |
q |
quantile of the prior on the error variance at which the data-based estimate is placed. |
nu |
regression degrees of freedom for the inverse X^2 prior. |
mh_prob_steps |
vector of prior probabilities for proposing changes to the tree structures: (GROW, PRUNE, CHANGE). |
verbose |
logical indicating whether to print progress information about the algorithm. |
... |
additional arguments to |
binary
, numeric
alpha
, beta
, k
, nu
Further model details can be found in the source link below.
In calls to varimp
for BARTMachineModel
, argument
metric
may be spedified as "splits"
(default) for the
proportion of time each predictor is chosen for a splitting rule or as
"trees"
for the proportion of times each predictor appears in a tree.
Argument num_replicates
is also available to control the number of
BART replicates used in estimating the inclusion proportions [default: 5].
Variable importance is automatically scaled to range from 0 to 100. To
obtain unscaled importance values, set scale = FALSE
. See example
below.
MLModel
class object.
1 2 |
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.