predict.softbart_regression {SoftBart}

Predict for SoftBart Regression

Description

Computes predictions from a softbart_regression object on new data.

Usage

## S3 method for class 'softbart_regression'
predict(object, newdata, iterations = NULL, ...)

Arguments

object

A softbart_regression object obtained as output of the softbart_regression() function.

newdata

A dataset to construct predictions on.

iterations

The iterations to get predictions on; includes all of iterations including burn-in and thinning iterations. Defaults to the saved iterations, running from (num_burn + num_thin):(num_burn + num_thin * num_save).

...

Other arguments passed to predict.

Value

A list containing

• mu: samples of the predicted value for each observation and iteration.

• mu_mean: posterior predicted values for each observation.