tidytreatment (version 0.3.1)

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

Description

Get fitted draws from posterior of bartMachine model

Usage

# S3 method for bartMachine
fitted_draws(
  model,
  newdata,
  value = ".value",
  ...,
  n = NULL,
  include_newdata = TRUE,
  include_sigsqs = FALSE
)

Value

A tidy data frame (tibble) with fitted values.

Arguments

model

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 fitted_draws; default ".value".

...

Not currently in use.

n

Not currently implemented.

include_newdata

Should the newdata be included in the tibble?

include_sigsqs

Should the posterior sigma-squared draw be included?