Makes predictions from a fitted MAIHDA model, either at the stratum level or individual level.
Usage
predict_maihda(object, newdata = NULL, type = c("individual", "strata"), ...)Arguments
- object
A maihda_model object from
fit_maihda().- newdata
Optional data frame for making predictions. If NULL, uses the original data from model fitting.
- type
Character string specifying prediction type:
"individual": Individual-level predictions including random effects
"strata": Stratum-level predictions (random effects only)
- ...
Additional arguments passed to predict method of underlying model.
Value
Depending on type:
For "individual": A numeric vector of predicted values
For "strata": A data frame with stratum ID and predicted random effect
Examples
# \donttest{
strata_result <- make_strata(maihda_sim_data, vars = c("gender", "race"))
model <- fit_maihda(health_outcome ~ age + (1 | stratum), data = strata_result$data)
# Individual predictions
pred_ind <- predict_maihda(model, type = "individual")
# Stratum predictions
pred_strata <- predict_maihda(model, type = "strata")
# }