treatment_effects: Get (individual) treatment effect draws from posterior

View source: R/treatment-effects-posterior.R

treatment_effectsR Documentation

Get (individual) treatment effect draws from 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

treatment_effects(
  model,
  treatment,
  newdata,
  subset = "all",
  common_support_method,
  cutoff,
  ...
)

Arguments

model

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

treatment

A character string specifying the name of the treatment variable.

newdata

Data frame to generate fitted values from. If omitted, defaults to the data used to fit the model.

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.

Value

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


tidytreatment documentation built on March 18, 2022, 6:30 p.m.