tidytreatment (version 0.3.1)

predicted_draws.wbart: Get predict draws from posterior of wbart model

Description

Get predict draws from posterior of wbart model

Usage

# S3 method for wbart
predicted_draws(
  object,
  newdata,
  value = ".prediction",
  ...,
  ndraws = NULL,
  include_newdata = TRUE,
  include_fitted = FALSE,
  include_sigsqs = FALSE
)

Value

A tidy data frame (tibble) with predicted values.

Arguments

object

A wbart 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".

...

Use to specify random number generator, default is rng=stats::rnorm.

ndraws

Not currently implemented.

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?