Obsługa brakujących danych z imputacją w R
Michal Oleszak
Machine Learning Engineer

Większość modeli statystycznych szacuje rozkład warunkowy zmiennej odpowiedzi:
$p(y|X)$
Aby uzyskać pojedynczą prognozę, rozkład warunkowy jest podsumowywany:
Można zamiast tego losować z tych rozkładów, aby zwiększyć zmienność.


Zadanie: imputacja PhysActive z danych nhanes za pomocą regresji logistycznej.
nhanes_imp <- hotdeck(nhanes)
missing_physactive <- is.na(nhanes$PhysActive)
Zadanie: imputacja PhysActive z danych nhanes za pomocą regresji logistycznej.
nhanes_imp <- hotdeck(nhanes)
missing_physactive <- is.na(nhanes$PhysActive)
logreg_model <- glm(PhysActive ~ Age + Weight + Pulse,
data = nhanes_imp, family = binomial)
Zadanie: imputacja PhysActive z danych nhanes za pomocą regresji logistycznej.
nhanes_imp <- hotdeck(nhanes)
missing_physactive <- is.na(nhanes$PhysActive)
logreg_model <- glm(PhysActive ~ Age + Weight + Pulse,
data = nhanes_imp, family = binomial)
preds <- predict(logreg_model, type = "response")
Zadanie: imputacja PhysActive z danych nhanes za pomocą regresji logistycznej.
nhanes_imp <- hotdeck(nhanes)
missing_physactive <- is.na(nhanes$PhysActive)
logreg_model <- glm(PhysActive ~ Age + Weight + Pulse,
data = nhanes_imp, family = binomial)
preds <- predict(logreg_model, type = "response")
preds <- ifelse(preds >= 0.5, 1, 0)
Zadanie: imputacja PhysActive z danych nhanes za pomocą regresji logistycznej.
nhanes_imp <- hotdeck(nhanes)
missing_physactive <- is.na(nhanes$PhysActive)
logreg_model <- glm(PhysActive ~ Age + Weight + Pulse,
data = nhanes_imp, family = binomial)
preds <- predict(logreg_model, type = "response")
preds <- ifelse(preds >= 0.5, 1, 0)
nhanes_imp[missing_physactive, "PhysActive"] <- preds[missing_physactive]
Zmienność imputowanych danych:
table(preds[missing_physactive])
1
26
Zmienność obserwowanych danych PhysActive:
table(nhanes$PhysActive)
0 1
181 610
nhanes_imp <- hotdeck(nhanes)
missing_physactive <- is.na(nhanes$PhysActive)
logreg_model <- glm(PhysActive ~ Age + Weight + Pulse,
data = nhanes_imp, family = binomial)
preds <- predict(logreg_model, type = "response")
preds <- ifelse(preds >= 0.5, 1, 0)
nhanes_imp[missing_physactive, "PhysActive"] <- preds[missing_physactive]
nhanes_imp <- hotdeck(nhanes)
missing_physactive <- is.na(nhanes$PhysActive)
logreg_model <- glm(PhysActive ~ Age + Weight + Pulse,
data = nhanes_imp, family = binomial)
preds <- predict(logreg_model, type = "response")
nhanes_imp[missing_physactive, "PhysActive"] <- preds[missing_physactive]
nhanes_imp <- hotdeck(nhanes)
missing_physactive <- is.na(nhanes$PhysActive)
logreg_model <- glm(PhysActive ~ Age + Weight + Pulse,
data = nhanes_imp, family = binomial)
preds <- predict(logreg_model, type = "response")
preds <- rbinom(length(preds), size = 1, prob = preds)
nhanes_imp[missing_physactive, "PhysActive"] <- preds[missing_physactive]
Zmienność imputowanych danych:
table(preds[missing_physactive])
0 1
5 21
Zmienność obserwowanych danych PhysActive:
table(nhanes$PhysActive)
0 1
181 610
Obsługa brakujących danych z imputacją w R