tidytreatment (version 0.3.1)

fitted_draws_BART: Get fitted draws from posterior of BART-package models

Description

Get fitted draws from posterior of BART-package models

Usage

fitted_draws_BART(
  model,
  newdata = NULL,
  value = ".value",
  ...,
  include_newdata = TRUE,
  include_sigsqs = FALSE,
  scale = "real"
)

Value

A tidy data frame (tibble) with fitted values.

Arguments

model

A model from BART package.

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

...

Arguments to pass to predict (e.g. BART:::predict.wbart).

include_newdata

Should the newdata be included in the tibble?

include_sigsqs

Should the posterior sigma-squared draw be included?

scale

Should the fitted values be on the real, probit or logit scale?