R/gather_emmeans_draws.R
gather_emmeans_draws.RdExtract draws from the result of a call to emmeans::emmeans() (formerly lsmeans)
or emmeans::ref_grid() applied to a Bayesian model.
gather_emmeans_draws(object, value = ".value", ...)
# Default S3 method
gather_emmeans_draws(object, value = ".value", ...)
# S3 method for class 'emm_list'
gather_emmeans_draws(object, value = ".value", grid = ".grid", ...)An emmGrid object such as returned by
emmeans::ref_grid() or emmeans::emmeans().
The name of the output column to use to contain the values of draws. Defaults to ".value".
Additional arguments passed to the underlying method for the type of object given.
If object is an emmeans::emm_list(), the name of the output column to use to contain the name of the
reference grid that a given row corresponds to. Defaults to ".grid".
A tidy data frame of draws. The columns of the reference grid are returned as-is, with an
additional column called .value (by default) containing marginal draws. The resulting data
frame is grouped by the columns from the reference grid to make use of summary functions like
point_interval() straightforward.
If object is an emmeans::emm_list(), which contains estimates from different reference grids,
an additional column with the default name of ".grid" is added to indicate the reference grid for each row in the output.
The name of this column is controlled by the grid argument.
emmeans::emmeans() provides a convenient syntax for generating draws from "estimated marginal means" from a model,
and can be applied to various Bayesian models, like rstanarm::stanreg-objects and
MCMCglmm::MCMCglmm(). Given a emmeans::ref_grid() object as returned by functions like
emmeans::ref_grid() or emmeans::emmeans() applied to a Bayesian model,
gather_emmeans_draws returns a tidy format data frame of draws from
the marginal posterior distributions generated by emmeans::emmeans().
# \dontrun{
library(dplyr)
library(magrittr)
library(brms)
library(emmeans)
#> Welcome to emmeans.
#> Caution: You lose important information if you filter this package's results.
#> See '? untidy'
# Here's an example dataset with a categorical predictor (`condition`) with several levels:
set.seed(5)
n = 10
n_condition = 5
ABC = tibble(
condition = rep(c("A","B","C","D","E"), n),
response = rnorm(n * 5, c(0,1,2,1,-1), 0.5)
)
m = brm(response ~ condition, data = ABC,
# 1 chain / few iterations just so example runs quickly
# do not use in practice
chains = 1, iter = 500)
#> Compiling Stan program...
#> Start sampling
#>
#> SAMPLING FOR MODEL 'anon_model' NOW (CHAIN 1).
#> Chain 1:
#> Chain 1: Gradient evaluation took 5.3e-05 seconds
#> Chain 1: 1000 transitions using 10 leapfrog steps per transition would take 0.53 seconds.
#> Chain 1: Adjust your expectations accordingly!
#> Chain 1:
#> Chain 1:
#> Chain 1: Iteration: 1 / 500 [ 0%] (Warmup)
#> Chain 1: Iteration: 50 / 500 [ 10%] (Warmup)
#> Chain 1: Iteration: 100 / 500 [ 20%] (Warmup)
#> Chain 1: Iteration: 150 / 500 [ 30%] (Warmup)
#> Chain 1: Iteration: 200 / 500 [ 40%] (Warmup)
#> Chain 1: Iteration: 250 / 500 [ 50%] (Warmup)
#> Chain 1: Iteration: 251 / 500 [ 50%] (Sampling)
#> Chain 1: Iteration: 300 / 500 [ 60%] (Sampling)
#> Chain 1: Iteration: 350 / 500 [ 70%] (Sampling)
#> Chain 1: Iteration: 400 / 500 [ 80%] (Sampling)
#> Chain 1: Iteration: 450 / 500 [ 90%] (Sampling)
#> Chain 1: Iteration: 500 / 500 [100%] (Sampling)
#> Chain 1:
#> Chain 1: Elapsed Time: 0.012 seconds (Warm-up)
#> Chain 1: 0.006 seconds (Sampling)
#> Chain 1: 0.018 seconds (Total)
#> Chain 1:
# Once we've fit the model, we can use emmeans() (and functions
# from that package) to get whatever marginal distributions we want.
# For example, we can get marginal means by condition:
m %>%
emmeans(~ condition) %>%
gather_emmeans_draws() %>%
median_qi()
#> # A tibble: 5 × 7
#> condition .value .lower .upper .width .point .interval
#> <fct> <dbl> <dbl> <dbl> <dbl> <chr> <chr>
#> 1 A 0.191 -0.128 0.542 0.95 median qi
#> 2 B 1.02 0.685 1.35 0.95 median qi
#> 3 C 1.87 1.54 2.21 0.95 median qi
#> 4 D 1.03 0.683 1.37 0.95 median qi
#> 5 E -0.948 -1.30 -0.584 0.95 median qi
# or we could get pairwise differences:
m %>%
emmeans(~ condition) %>%
contrast(method = "pairwise") %>%
gather_emmeans_draws() %>%
median_qi()
#> # A tibble: 10 × 7
#> contrast .value .lower .upper .width .point .interval
#> <chr> <dbl> <dbl> <dbl> <dbl> <chr> <chr>
#> 1 A - B -0.834 -1.27 -0.305 0.95 median qi
#> 2 A - C -1.67 -2.15 -1.18 0.95 median qi
#> 3 A - D -0.823 -1.29 -0.360 0.95 median qi
#> 4 A - E 1.13 0.646 1.70 0.95 median qi
#> 5 B - C -0.862 -1.38 -0.368 0.95 median qi
#> 6 B - D 0.000319 -0.526 0.510 0.95 median qi
#> 7 B - E 1.97 1.44 2.47 0.95 median qi
#> 8 C - D 0.865 0.349 1.31 0.95 median qi
#> 9 C - E 2.82 2.34 3.28 0.95 median qi
#> 10 D - E 1.98 1.50 2.42 0.95 median qi
# see the documentation of emmeans() for more examples of types of
# contrasts supported by that packge.
# }