compose_data {tidybayes} | R Documentation |
Compose data for input into a Bayesian model
Description
Compose data into a list suitable to be passed into a Bayesian model (JAGS, BUGS, Stan, etc).
Usage
compose_data(..., .n_name = n_prefix("n"))
Arguments
... |
Data to be composed into a list suitable for being passed into
Stan, JAGS, etc. Named arguments will have their name used as the |
.n_name |
A function that is used to form dimension index variables (a variable
whose value is number of levels in a factor or the length of a data frame in
|
Details
This function recursively translates each argument into list elements using
as_data_list()
, merging all resulting lists together. By
default this means that:
numerics are included as-is.
logicals are translated into numeric using
as.numeric()
.factors are translated into numeric using
as.numeric()
, and an additional element named.n_name(argument_name)
is added with the number of levels in the factor. The default.n_name
function prefixes"n_"
before the factor name; e.g. a factor namedfoo
will have an element namedn_foo
added containing the number of levels infoo
.character vectors are converted into factors then translated into numeric in the same manner as factors are.
lists are translated by translating all elements of the list (recursively) and adding them to the result.
data.frames are translated by translating every column of the data.frame and adding them to the result. A variable named
"n"
(or.n_name(argument_name)
if the data.frame is passed as a named argumentargument_name
) is also added containing the number of rows in the data frame.-
NULL
values are dropped. Setting a named argument toNULL
can be used to drop that item from the resulting list (if an unwanted element was added to the list by a previous argument, such as a column from a data frame that is not needed in the model). all other types are dropped (and a warning given)
As in functions like mutate()
, each expression is evaluated in an
environment containing the data list built up so far.
For example, this means that if the first argument to compose_data
is a data frame, subsequent arguments can include direct references to columns
from that data frame. This allows you, for example, to easily use
x_at_y()
to generate indices for nested models.
If you wish to add support for additional types not described above,
provide an implementation of as_data_list()
for the type. See
the implementations of as_data_list.numeric
,
as_data_list.logical
, etc for examples.
Value
A list where each element is a translated variable as described above.
Author(s)
Matthew Kay
See Also
x_at_y()
, spread_draws()
,
gather_draws()
.
Examples
library(magrittr)
df = data.frame(
plot = factor(paste0("p", rep(1:8, times = 2))),
site = factor(paste0("s", rep(1:4, each = 2, times = 2)))
)
# without changing `.n_name`, compose_data() will prefix indices
# with "n" by default
df %>%
compose_data()
# you can use n_prefix() to define a different prefix (e.g. "N"):
df %>%
compose_data(.n_name = n_prefix("N"))
# If you have nesting, you may want a nested index, which can be generated using x_at_y()
# Here, site[p] will give the site for plot p
df %>%
compose_data(site = x_at_y(site, plot))