predicted_draws_BART: Get predict draws from posterior of 'BART'-package models

In tidytreatment: Tidy Methods for Bayesian Treatment Effect Models

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.

tidytreatment documentation built on March 18, 2022, 6:30 p.m.