bartMachine (version 1.2.6)

get_sigsqs: Get Posterior Error Variance Estimates

Description

Returns the posterior estimates of the error variance from the Gibbs samples with an option to create a histogram of the posterior estimates of the error variance with a credible interval overlaid.

Usage

get_sigsqs(bart_machine, after_burn_in = T, 
plot_hist = F, plot_CI = .95, plot_sigma = F)

Value

Returns a vector of posterior \(\sigma^2\) draws (with or without the burn-in samples).

Arguments

bart_machine

An object of class ``bartMachine''.

after_burn_in

If TRUE, only the \(\sigma^2\) draws after the burn-in period are returned.

plot_hist

If TRUE, a histogram of the posterior \(\sigma^2\) draws is generated.

plot_CI

Confidence level for credible interval on histogram.

plot_sigma

If TRUE, plots \(\sigma\) instead of \(\sigma^2\).

Author

Adam Kapelner and Justin Bleich

See Also

get_sigsqs

Examples

Run this code
if (FALSE) {
#generate Friedman data
set.seed(11)
n  = 300 
p = 5
X = data.frame(matrix(runif(n * p), ncol = p))
y = 10 * sin(pi* X[ ,1] * X[,2]) +20 * (X[,3] -.5)^2 + 10 * X[ ,4] + 5 * X[,5] + rnorm(n)

##build BART regression model
bart_machine = bartMachine(X, y)

#get posterior sigma^2's after burn-in and plot
sigsqs = get_sigsqs(bart_machine, plot_hist = TRUE)
}

Run the code above in your browser using DataCamp Workspace