Learn R Programming

tidytreatment (version 0.3.1)

predicted_draws.bartcFit: Get prediction draws from posterior of bartCause-package objects

Description

Get prediction draws from posterior of bartCause-package objects

Usage

# S3 method for bartcFit
predicted_draws(
  object,
  ...,
  value = ".prediction",
  re_formula = NULL,
  fitstage = c("response", "assignment")
)

Arguments

object

A bartCauseFit object.

...

Additional arguments (e.g. newdata) passed to the underlying prediction method for the type of model given.

value

The name of the output column.

re_formula

If NULL (default), include all group-level effects; if NA, include no group-level effects.

fitstage

If is.null(type), return posterior from response or treatment assignment model.