tidytreatment0.3.1 package

Tidy Methods for Bayesian Treatment Effect Models

avg_treatment_effects

Get (conditional) average treatment effect draws from posterior

covariate_importance

Counts of variable overall inclusion

covariate_with_treatment_importance

Counts of variable inclusion when interacting with treatment

epred_draws.bartcFit

Get expected prediction draws from posterior of bartCause-package ob...

epred_draws.stan4bartFit

Get expected prediction draws from posterior of stan4bart-package mo...

fitted_draws_BART

Get fitted draws from posterior of BART-package models

fitted_draws.bartMachine

Get fitted draws from posterior of bartMachine model

fitted_draws.lbart

Get fitted draws from posterior of lbart model

fitted_draws.mbart

Get fitted draws from posterior of mbart model

fitted_draws.mbart2

Get fitted draws from posterior of mbart2 model

fitted_draws.pbart

Get fitted draws from posterior of pbart model

fitted_draws.wbart

Get fitted draws from posterior of wbart model

has_common_support

Evaluate if observations have common support.

has_tidytreatment_methods

Check if a model class has required generic methods for tidytreatment ...

linpred_draws.bartcFit

Get expected prediction draws (on linear scale) from posterior of `bar...

linpred_draws.stan4bartFit

Get expected prediction draws (on linear scale) from posterior of `sta...

pipe

Pipe operator

posterior_trees_BART

Get posterior tree draws into tibble format from BART model

predicted_draws_BART

Get predict draws from posterior of BART-package models

predicted_draws.bartcFit

Get prediction draws from posterior of bartCause-package objects

predicted_draws.bartMachine

Get predict draws from posterior of bartMachine model

predicted_draws.stan4bartFit

Get prediction draws from posterior of stan4bart-package models

predicted_draws.wbart

Get predict draws from posterior of wbart model

residual_draws_BART

Get residual draw for BART model

residual_draws.bartMachine

Get residual draw for bartMachine model

residual_draws.pbart

Get residual draw for pbart model

residual_draws.wbart

Get residual draw for wbart model

simulate_su_hill_data

Simulate data with scenarios from Hill and Su (2013)

tidy_ate

Get average treatment effect draws from posterior

tidy_att

Get average treatment effect on treated draws from posterior

tidy_draws.bartcFit

Tidy access to posterior of bartCause-package objects

tidytreatment

tidytreatment: Tidy methods for Bayesian treatment effect models

treatment_effects.bartcFit

Get (individual) treatment effect draws from bartcFit posterior

treatment_effects.default

Get treatment effect draws from posterior

treatment_effects

Get (individual) treatment effect draws from posterior

variance_draws

Get variance draws from posterior of BART models

Functions for extracting tidy data from Bayesian treatment effect models, in particular BART, but extensions are possible. Functionality includes extracting tidy posterior summaries as in 'tidybayes' <https://github.com/mjskay/tidybayes>, estimating (average) treatment effects, common support calculations, and plotting useful summaries of these.

  • Maintainer: Joshua J Bon
  • License: MIT + file LICENSE
  • Last published: 2025-01-10