Modifying plots

plotBart uses ggplot2 to construct plots. These functions return ggplot objects that can be manipulated similar to standard ggplot2::ggplot() calls. Plot titles, subtitles, labels, captions, and themes can be easily customized using ggplot syntax.

library(plotBart)
data(lalonde)
confounders <- c('age', 'educ', 'black', 'hisp', 'married', 'nodegr')

# plot balance across treatment and control groups
p <- plot_balance(.data = lalonde,
                  treatment = 'treat',
                  confounders = confounders)
p


Modify the plot labels:

p +
  labs(title = 'My comments on the results',
       subtitle = NULL,
       caption = 'December 2021',
       x = 'Mean diff b/t treatment and control')


Or change the theme entirely:

p + 
  theme_classic()

# set the theme for all plots within this R session
theme_set(theme_bw())


Extract data to reconstruct the plot:

p$data
#> # A tibble: 6 × 2
#>   name       diff
#>   <chr>     <dbl>
#> 1 age      0.107 
#> 2 black    0.0438
#> 3 educ     0.144 
#> 4 hisp    -0.170 
#> 5 married  0.0943
#> 6 nodegr  -0.306


Refer to the ggplot2 documentation for more details: