data-raw/categorical_example.R

In santikka/dynamite: Bayesian Modeling and Causal Inference for Multivariate Longitudinal Data

set.seed(1)
n_id <- 100L
n_time <- 20L

d <- data.frame(
  y = sample(factor(c("a", "b", "c")), size = n_id, replace = TRUE),
  x = sample(factor(c("A", "B", "C")), size = n_id, replace = TRUE),
  time = 1,
  id = seq_len(n_id)
)

d <- dplyr::right_join(
  d,
  data.frame(
    time = rep(seq_len(n_time), each = n_id),
    id = seq_len(n_id)
  )
)

d$z <- rnorm(nrow(d))

f <- obs(x ~ z + lag(x) + lag(y), family = "categorical") +
  obs(y ~ z + lag(x) + lag(y), family = "categorical")

init <- list(
  beta_x_B = c(2, 0.8, 0.2, 0, 0),
  beta_x_C = c(1, 0.5, 2, 0.2, 0.1),
  a_x_B = -0.1,
  a_x_C = 0.2,
  beta_y_b = c(0, 1, 0.8, 0.3, 0.5),
  beta_y_c = c(1, 0.2, 0, 0.3, -0.5),
  a_y_b = 0.1,
  a_y_c = -0.5
)

fit <- dynamite(
  dformula = f,
  data = d,
  time = "time",
  group = "id",
  chains = 1,
  iter = 1,
  algorithm = "Fixed_param",
  init = list(init),
)

categorical_example <- predict(fit, type = "response") |>
  dplyr::mutate(x = x_new, y = y_new) |>
  dplyr::select(id, time, x, y, z)

usethis::use_data(categorical_example, overwrite = TRUE)