Package: tidytreatment 0.3.2.9000

tidytreatment: Tidy Methods for Bayesian Treatment Effect 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.

Authors:Joshua J Bon [aut, cre]

tidytreatment_0.3.2.9000.tar.gz
tidytreatment_0.3.2.9000.zip(r-4.7)tidytreatment_0.3.2.9000.zip(r-4.6)tidytreatment_0.3.2.9000.zip(r-4.5)
tidytreatment_0.3.2.9000.tgz(r-4.6-any)tidytreatment_0.3.2.9000.tgz(r-4.5-any)
tidytreatment_0.3.2.9000.tar.gz(r-4.7-any)tidytreatment_0.3.2.9000.tar.gz(r-4.6-any)
tidytreatment_0.3.2.9000.tgz(r-4.6-emscripten)
manual.pdf |manual.html✨
card.svg |card.png
tidytreatment/json (API)
NEWS

# Install 'tidytreatment' in R:
install.packages('tidytreatment', repos = c('https://bonstats.r-universe.dev', 'https://cloud.r-project.org'))

Bug tracker:https://github.com/bonstats/tidytreatment/issues

Datasets:

On CRAN:

Conda:

bayesiantidytreatment-effects

6.45 score 18 stars 13 scripts 246 downloads 11 exports 60 dependencies

Last updated from:5fa4fac789. Checks:9 OK. Indexed: yes.

TargetResultTimeFilesSyslog
linux-devel-x86_64OK190
source / vignettesOK363
linux-release-x86_64OK181
macos-release-arm64OK150
macos-oldrel-arm64OK205
windows-develOK182
windows-releaseOK189
windows-oldrelOK169
wasm-releaseOK149

Exports:%>%avg_treatment_effectscovariate_importancecovariate_with_treatment_importancehas_common_supporthas_tidytreatment_methodssimulate_su_hill_datatidy_atetidy_atttreatment_effectsvariance_draws

Dependencies:abindarrayhelpersbackportsBARTbitbit64checkmateclicliprcodacpp11crayondbartsdistributionaldplyrfarvergenericsggdistggplot2gluegtablehmsisobandlabelinglatticelifecyclemagrittrMatrixmatrixStatsnlmenumDerivpillarpkgconfigposteriorprettyunitsprogresspurrrquadprogR6RColorBrewerRcppreadrrlangS7scalesstringistringrsurvivalsvUnittensorAtibbletidybayestidyrtidyselecttzdbutf8vctrsviridisLitevroomwithr

Using the tidytreatment package with BART

Joshua J Bon

Rendered fromuse-tidytreatment-BART.Rmdusingknitr::rmarkdownon May 07 2026.

Last update: 2025-12-05
Started: 2020-04-27

Using the tidytreatment package with BART (binary outcome)

Joshua J Bon

Rendered fromuse-tidytreatment-BART-binary-outcome.Rmdusingknitr::rmarkdownon May 07 2026.

Last update: 2025-12-05
Started: 2025-01-11

Using the tidytreatment package with bartCause

Joshua J Bon

Rendered fromuse-tidytreatment-bartCause.Rmdusingknitr::rmarkdownon May 07 2026.

Last update: 2025-12-05
Started: 2023-05-18

Readme and manuals

Help Manual

Help pageTopics
Get (conditional) average treatment effect draws from posterioravg_treatment_effects
Example model 1bartmodel1
Model matrix used for 'bartmodel1'bartmodel1_modelmatrix
Counts of variable overall inclusioncovariate_importance
Counts of variable inclusion when interacting with treatmentcovariate_with_treatment_importance
Get fitted draws from posterior of 'BART'-package modelsepred_draws_BART
Get expected prediction draws from posterior of 'bartCause'-package objectsepred_draws.bartcFit
Get fitted draws from posterior of 'bartMachine' modelepred_draws.bartMachine
Get fitted draws from posterior of 'lbart' modelepred_draws.lbart
Get fitted draws from posterior of 'mbart' modelepred_draws.mbart
Get fitted draws from posterior of 'mbart2' modelepred_draws.mbart2
Get fitted draws from posterior of 'pbart' modelepred_draws.pbart
Get expected prediction draws from posterior of 'stan4bart'-package modelsepred_draws.stan4bartFit
Get fitted draws from posterior of 'wbart' modelepred_draws.wbart
Evaluate if observations have common support.has_common_support
Check if a model class has required generic methods for tidytreatment functions.has_tidytreatment_methods
ACIC2019 High Dimensional Test DatasethighDim_testdataset3
Get expected prediction draws (on linear scale) from posterior of 'bartCause'-package objectslinpred_draws.bartcFit
Get linear predictor draws from posterior of 'stan4bart'-package modelslinpred_draws.stan4bartFit
Get posterior tree draws into tibble format from BART modelposterior_trees_BART
Get predict draws from posterior of 'BART::wbart' modelspredicted_draws_BART
Get prediction draws from posterior of 'bartCause'-package objectspredicted_draws.bartcFit
Get predict draws from posterior of 'bartMachine' modelpredicted_draws.bartMachine
Get predict draws from posterior of 'lbart' modelpredicted_draws.lbart
Get predict draws from posterior of 'pbart' modelpredicted_draws.pbart
Get prediction draws from posterior of 'stan4bart'-package modelspredicted_draws.stan4bartFit
Get predict draws from posterior of 'wbart' modelpredicted_draws.wbart
Get residual draw for BART modelresidual_draws_BART
Get residual draw for 'bartMachine' modelresidual_draws.bartMachine
Get residual draw for 'pbart' modelresidual_draws.pbart
Get residual draw for 'wbart' modelresidual_draws.wbart
Simulate data with scenarios from Hill and Su (2013)simulate_su_hill_data
Example simulated dataset 1suhillsim1
Example simulated dataset 2: with subject specific random effectssuhillsim2_ranef
Get average treatment effect draws from posteriortidy_ate
Get average treatment effect on treated draws from posteriortidy_att
Tidy access to posterior of 'bartCause'-package objectstidy_draws.bartcFit
tidytreatment: Tidy methods for Bayesian treatment effect modelstidytreatment-package tidytreatment
Get (individual) treatment effect draws from posteriortreatment_effects
Get (individual) treatment effect draws from bartcFit posteriortreatment_effects.bartcFit
Get treatment effect draws from posteriortreatment_effects.default
Get variance draws from posterior of BART modelsvariance_draws