Learn R Programming

tidytreatment (version 0.3.1)

linpred_draws.bartcFit: Get expected prediction draws (on linear scale) from posterior of bartCause-package objects

Description

Typically referred to as fitted value draws on linear scale, where appropriate.

Usage

# S3 method for bartcFit
linpred_draws(
  object,
  ...,
  value = ".linpred",
  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.