epred_draws.bartcFit {tidytreatment}R Documentation

Get expected prediction draws from posterior of bartCause-package objects

Description

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

Usage

## S3 method for class 'bartcFit'
epred_draws(
  object,
  ...,
  value = ".epred",
  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.


[Package tidytreatment version 0.3.1 Index]