tidytreatment (version 0.3.1)

treatment_effects.bartcFit: Get (individual) treatment effect draws from bartcFit posterior

Description

CTE = Conditional Treatment Effects (usually used to generate (C)ATE or ATT) newdata specifies the conditions, if unspecified it defaults to the original data. Assumes treated column is either a integer column of 1's (treated) and 0's (nontreated) or logical indicating treatment if TRUE.

Usage

# S3 method for bartcFit
treatment_effects(
  model,
  treatment = NULL,
  newdata = NULL,
  subset = "all",
  common_support_method,
  cutoff,
  ...
)

Value

A tidy data frame (tibble) with treatment effect values.

Arguments

model

A supported Bayesian model fit that can provide fits and predictions.

treatment

Not used. Treatment variable specified by bartcFit object.

newdata

Not used. extracts treatment effects already calculated by bartcFit object.

subset

Either "treated", "nontreated", or "all". Default is "all".

common_support_method

Either "sd", or "chisq". Default is unspecified, and no common support calculation is done.

cutoff

Cutoff for common support (if in use).

...

Arguments to be passed to tidybayes::fitted_draws typically scale for BART models.