Learn R Programming
tidytreatment (version 0.3.2)

epred_draws.bartMachine: Get fitted draws from posterior of bartMachine model

Description

Get fitted draws from posterior of bartMachine model

Usage

# S3 method for bartMachine
epred_draws(
  object,
  newdata,
  value = ".value",
  ...,
  ndraws = NULL,
  include_newdata = TRUE,
  include_sigsqs = FALSE
)

Value

A tidy data frame (tibble) with fitted values.

Arguments

object

A bartMachine model.

newdata

Data frame to generate fitted values from. If omitted, defaults to the data used to fit the model.

value

The name of the output column for epred_draws; default ".value".

...

Not currently in use.

ndraws

Not currently implemented.

include_newdata

Should the newdata be included in the tibble?

include_sigsqs

Should the posterior sigma-squared draw be included?