stanvar {brms} | R Documentation |
User-defined variables passed to Stan
Description
Prepare user-defined variables to be passed to one of Stan's program blocks. This is primarily useful for defining more complex priors, for refitting models without recompilation despite changing priors, or for defining custom Stan functions.
Usage
stanvar(
x = NULL,
name = NULL,
scode = NULL,
block = "data",
position = "start",
pll_args = NULL
)
Arguments
x |
An R object containing data to be passed to Stan.
Only required if |
name |
Optional character string providing the desired variable
name of the object in |
scode |
Line of Stan code to define the variable
in Stan language. If |
block |
Name of one of Stan's program blocks in
which the variable should be defined. Can be |
position |
Name of the position within the block where the
Stan code should be placed. Currently allowed are |
pll_args |
Optional Stan code to be put into the header
of |
Details
The stanvar
function is not vectorized. Instead, multiple
stanvars
objects can be added together via +
(see Examples).
Special attention is necessary when using stanvars
to inject
code into the 'likelihood'
block while having threading
activated. In this case, your custom Stan code may need adjustments to ensure
correct observation indexing. Please investigate the generated Stan code via
stancode
to see which adjustments are necessary in your case.
Value
An object of class stanvars
.
Examples
bprior <- prior(normal(mean_intercept, 10), class = "Intercept")
stanvars <- stanvar(5, name = "mean_intercept")
stancode(count ~ Trt, epilepsy, prior = bprior,
stanvars = stanvars)
# define a multi-normal prior with known covariance matrix
bprior <- prior(multi_normal(M, V), class = "b")
stanvars <- stanvar(rep(0, 2), "M", scode = " vector[K] M;") +
stanvar(diag(2), "V", scode = " matrix[K, K] V;")
stancode(count ~ Trt + zBase, epilepsy,
prior = bprior, stanvars = stanvars)
# define a hierachical prior on the regression coefficients
bprior <- set_prior("normal(0, tau)", class = "b") +
set_prior("target += normal_lpdf(tau | 0, 10)", check = FALSE)
stanvars <- stanvar(scode = "real<lower=0> tau;",
block = "parameters")
stancode(count ~ Trt + zBase, epilepsy,
prior = bprior, stanvars = stanvars)
# ensure that 'tau' is passed to the likelihood of a threaded model
# not necessary for this example but may be necessary in other cases
stanvars <- stanvar(scode = "real<lower=0> tau;",
block = "parameters", pll_args = "real tau")
stancode(count ~ Trt + zBase, epilepsy,
stanvars = stanvars, threads = threading(2))