Предсказать после множественного вменения в R
Я использовал пакет мышей в R, чтобы выполнить многократное вменение для моих данных:
### multiple inputation by chained equations
imp.data <- mice(data, maxit = 5, m = 5, seed = 92385, print = F)
Я хочу запустить модель логистической регрессии после ИМ и прогнозировать результат на основе этой модели:
model <- with(imp.data, glm(died ~ agecat + female_1 + insurance + mech + transfer +
iss + mxaisbr1 + maxais + cm_chf_1 + cm_mets_1 + cm_liver_1 +
cm_htn_c_1 + cm_bldloss_1 + state, family = binomial))
Тем не менее, команда прогнозирования не работает:
predict(pool(model), type = c('response'))
Было бы намного проще, если бы у меня были данные с вмененными значениями, но вменение получило 5 вмененных наборов данных, что усложнило оценку после публикации. Любая идея?
Спасибо!!
1 ответ
Я не уверен, что вмененные модели сохраняются как data.frame
или матрица..
Но если вы преобразуете модель в data.frame
Вы можете подключить столбцы, представляющие интерес для вашего оригинального кадра.
imp.data <- data.frame(imp.data)
original.df$NewImputtedColumn <- imp.data[, 1] # Assuming you want column 1
Теперь вы можете легко сохранять копии представленных моделей и по-прежнему работать только с одним data.frame
в ваших прогнозных моделях.
Это то, что я склонен делать в любом случае, может быть не стандартным способом (я не уверен).
Вы берете только введенные значения от 1 из вмененных моделей, верно?