Description Usage Arguments Value Examples
View source: R/predict_bartBMA.R
This function produces predictions for a new dataset using a previously obtained bartBMA object.
1 | predict_bartBMA(object, newdata)
|
object |
A bartBMA object obtained using the barBMA function. |
newdata |
Covariate matrix for new dataset. |
A vector of predictions for the new dataset.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | set.seed(100)
#simulate some data
N <- 100
p<- 100
epsilon <- rnorm(N)
xcov <- matrix(runif(N*p), nrow=N)
y <- sin(pi*xcov[,1]*xcov[,2]) + 20*(xcov[,3]-0.5)^2+10*xcov[,4]+5*xcov[,5]+epsilon
epsilontest <- rnorm(N)
xcovtest <- matrix(runif(N*p), nrow=N)
ytest <- sin(pi*xcovtest[,1]*xcovtest[,2]) + 20*(xcovtest[,3]-0.5)^2+10*xcovtest[,4]+
5*xcovtest[,5]+epsilontest
#Train the object
bart_bma_example <- bartBMA(x.train = xcov,y.train=y,x.test=xcovtest,zero_split = 1,
only_max_num_trees = 1,split_rule_node = 0)
#Obtain the prediction intervals
predict_bartBMA(bart_bma_example,newdata=xcovtest)
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.