Learn R Programming
tidytreatment (version 0.3.1)

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

Description

Get predict draws from posterior of bartMachine model

Usage

# S3 method for bartMachine
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 bartMachine 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".

...

Not currently in use.

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?