search

idem-lab / epiwave.pipelines

0 stars 3 forks source link

allow fit_model to take in inits #51

Open AugustHao opened 1 month ago

AugustHao commented 1 month ago

add an argument so that one can provide init to the mcmc function inside the fit_model wrapper

AugustHao commented 1 month ago

add a wrapper to set up inits necessary for growth_rate_deriv option of the infection model

z_raw <- attr(infection_model_objects$gp, "gp_info")$v
n_chains <- 10
inits <- replicate(n_chains,
                   initials(
                     z_raw = rep(0, length(z_raw))
                   ),
                   simplify = FALSE)