generate_STb_model_TADA() — generate_STb_model

Helper function that creates stan code for cTADA and dTADA type models.

Usage

generate_STb_model_TADA(
  STb_data,
  model_type = "full",
  intrinsic_rate = "constant",
  transmission_func = "standard",
  dTADA = F,
  veff_params = c(),
  veff_type = "id",
  gq = TRUE,
  est_acqTime = FALSE,
  priors = list()
)

Arguments

STb_data: a list of formatted data returned from the STbayes_data() function
model_type: string specifying the model type: "asocial" or "full"
intrinsic_rate: string specifying whether intrinsic rate is static or can change over time ("constant", "weibull")
transmission_func: string specifying transmission function: "standard", "freqdep_f" or "freqdep_k" for complex contagion. Defaults to "standard". Complex contagion with multi-network model is not supported.
dTADA: boolean indicating whether dTADA should be used.
veff_params: Vector of parameter names (string) for which to estimate varying effects. Default is no varying effects.
veff_type: string/vector specifying whether varying effects should be applied to "id", "trial" or both (c("id","trial")). Default is id, applied only if user also gives veff_params.
gq: Boolean to indicate whether the generated quantities block is added (incl. ll for WAIC)
est_acqTime: Boolean to indicate whether gq block includes estimates for acquisition time. At the moment this uses 'one weird trick' to accomplish this and does not support estimates for non-integer learning times.
priors: named list with strings containing the prior for log baserate, s, f, k.

Value

A STAN model (character) that is customized to the input data.