Description Usage Arguments Value
Evaluates a Bayesian Additive Regression Tree models using the dbarts
package for n_boot
bootstrap iterations to obtain confidence bands for the estimated PDP. The dbarts
package estimates BART models much faster than BayesTree
.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
formula |
is an output formula for the outcome. |
variable |
is the treatment variable. |
data |
is a data frame to be used for the training. |
newdata |
is an optional data frame of test data for the PDPs. |
grid |
sets the values of |
outcome |
is the outcome class to be predicted for classification problems. |
n_boot |
is the number of bootstrap replications. |
p_boot |
is the proportion of the data to select for each bootstrap replication. |
N |
is the number of observations to select for calculating the PDPs. |
label |
is a character-string variable label for |
clock |
is a logical indicating whether to time each bootstrap replication. |
test |
is a logical indicating whether to calculate the pdp for both the |
seed |
is a random seed (default is 8675309). |
... |
additional arguments for the |
bsPDPdbarts
returns an object with class "bsPDP," a list that includes the following components:
variable |
the treatment variable. |
pdpData |
the estimated average predictions and standard errors along |
trainData |
the original training data. |
testData |
the test data. |
outcome |
the outcome class ( |
trControl |
is not applicable for this method ( |
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.