fitted_draws.bartMachine {tidytreatment}

Get fitted draws from posterior of bartMachine model

Description

Get fitted draws from posterior of bartMachine model

Usage

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

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?

Value

A tidy data frame (tibble) with fitted values.