predicted_draws_BART {tidytreatment}

Get predict draws from posterior of BART-package models

Description

Get predict draws from posterior of BART-package models

Usage

predicted_draws_BART(
  object,
  newdata = NULL,
  value = ".prediction",
  ...,
  rng = stats::rnorm,
  include_newdata = TRUE,
  include_fitted = FALSE,
  include_sigsqs = FALSE
)

Arguments

object

A BART-package model.

newdata

Data frame to generate predictions from. If omitted, most model types will generate predictions from the data used to fit the model.

value

The name of the output column for predicted_draws; default ".prediction".

...

Arguments to pass to predict (e.g. BART:::predict.wbart).

rng

Random number generator function. Default is rnorm for models with Gaussian errors.

include_newdata

Should the newdata be included in the tibble?

include_fitted

Should the posterior fitted values be included in the tibble?

include_sigsqs

Should the posterior sigma-squared draw be included?

Value

A tidy data frame (tibble) with predicted values.