plotBart is a diagnostic and plotting package forbartCause andthinkCausal.
library(plotBart)data(lalonde)confounders<-c('age','educ','black','hisp','married','nodegr')# fit BART modelmodel_results<- bartCause::bartc(response = lalonde[['re78']],treatment = lalonde[['treat']],confounders =as.matrix(lalonde[, confounders]),estimand ='ate',commonSup.rule ='none',verbose =FALSE,keepTrees =TRUE)# plot common supportplot_common_support(.model = model_results)
# plot CATE and manipulate ggplot objectplot_CATE(.model = model_results,type ='density',ci_80 =TRUE,ci_95 =TRUE,.mean =TRUE)+labs(subtitle ='My comments on the results')+theme_classic()
plotBart is currently in development and is available to test byinstalling via:
# install.packages("remotes")remotes::install_github('priism-center/plotBart@*release')