What if you want to predict the missing values instead of doing inference with incomplete data?
MICE
@amices.bsky.social
24 followers 4 following 4 posts
Home of the #RStats imputation package {mice}. Amices is a place for people interested in solving missing data problems. See amices.org & github.com/amices. Posts by @oberman.bsky.social
![Image of R code. To reproduce:
library(ggplot2)
library(dplyr)
library(mice, warn.conflicts = FALSE)
imp <- mice(nhanes, m = 5, maxit = 5, seed = 1,
ignore = rep(c(FALSE, TRUE), c(20, 5)),
print = FALSE)
impdats <- complete(imp, "all")
train <- lapply(impdats, function(dat) subset(dat, !imp$ignore))
test <- lapply(impdats, function(dat) subset(dat, imp$ignore))
fits <- lapply(train, function(dat) lm(age ~ bmi + hyp + chl, data = dat))
preds <- predict_mi(object = fits, newdata = test, pool = TRUE, interval = "prediction")
preds
preds %>%
as.data.frame() %>%
mutate(case = 1:nrow(preds),
y = test[[1]]$age) %>%
ggplot(aes(x = fit, y = case, col = rowSums(is.na(nhanes[imp$ignore,]))>0)) +
geom_point() +
geom_errorbar(aes(xmin = lwr, xmax = upr)) +
theme_minimal() +
scale_color_manual(values = mice::mdc(1:2), labels = c("observed", "missing")) +
theme(legend.title = element_blank(),
legend.position = "bottom") +
labs(x = "prediction",
title = "Pooled prediction intervals")](https://cdn.bsky.app/img/feed_thumbnail/plain/did:plc:omgr6y6c62b4qar7fganjtdy/bafkreiesbac4qo25mh5rkegnw4ntcvbwedvpzqms4zn7xiothhzlgchxcm)
