tests/simulation/Covariates.R

In bage: Bayesian Estimation and Forecasting of Age-Specific Rates

library(bage)
set.seed(0)

data <- expand.grid(age = poputils::age_labels(type = "lt"),
                    sex = c("Female", "Male"),
                    time = 2001:2015)
data$population <- runif(n = nrow(data), min = 100, max = 300)
data$deaths <- NA
data$income <- rnorm(n = nrow(data))
data$is_2024_male <- data$age == 2024 & data$sex == "Male"
mod_est <- mod_pois(deaths ~ age : sex + sex * time,
                    data = data,
                    exposure = population) |>
  set_prior(`(Intercept)` ~ NFix(s = 0.01)) |>
  set_prior(age:sex ~ SVD(HMD)) |>
  set_prior(time ~ RW(s = 0.05)) |>
  set_prior(sex:time ~ RW(s = 0.01)) |>
  set_prior(sex ~ NFix(sd = 0.1)) |>
  set_disp(mean = 0.05) |>
  set_covariates(~ income + is_2024_male)
rep <- report_sim(mod_est = mod_est, n_sim = 100, n_core = 10)
print(rep)