knitr::opts_chunk$set(echo = TRUE, comment = NA, error = TRUE)
Source: https://www.r-bloggers.com/missing-value-treatment/ http://r-statistics.co/Missing-Value-Treatment-With-R.html
Replacing the missing values with the mean / median / mode is a crude way of treating missing values. Depending on the context, like if the variation is low or if the variable has low leverage over the response, such a rough approximation is acceptable and could possibly give satisfactory results.
# initialize the data data ("BostonHousing", package="mlbench") original <- BostonHousing # backup original data # Introduce missing values set.seed(100) BostonHousing[sample(1:nrow(BostonHousing), 40), "rad"] <- NA BostonHousing[sample(1:nrow(BostonHousing), 40), "ptratio"] <- NA original.NA <- BostonHousing
The missing values have been injected. Though we know where the missings are, lets quickly check the ‘missings’ pattern using mice::md.pattern.
# Pattern of missing values library(mice) md.pattern(BostonHousing) # pattern or missing values in data.
There are really four ways you can handle missing values:
If you have large number of observations in your dataset, where all the classes to be predicted are sufficiently represented in the training data, then try deleting (or not to include missing values while model building, for example by setting na.action=na.omit) those observations (rows) that contain missing values. Make sure after deleting the observations, you have:
# Example lm(medv ~ ptratio + rad, data=BostonHousing, na.action=na.omit)
If a particular variable is having more missing values that rest of the variables in the dataset, and, if by removing that one variable you can save many observations. I would, then, suggest to remove that particular variable, unless it is a really important predictor that makes a lot of business sense. It is a matter of deciding between the importance of the variable and losing out on a number of observations.
Replacing the missing values with the mean / median / mode is a crude way of treating missing values. Depending on the context, like if the variation is low or if the variable has low leverage over the response, such a rough approximation is acceptable and could possibly give satisfactory results.
library(Hmisc) impute(BostonHousing$ptratio, mean) # replace with mean
impute(BostonHousing$ptratio, median) # median
impute(BostonHousing$ptratio, 20) # replace specific number
summary(BostonHousing$ptratio)
# or if you want to impute manually BostonHousing$ptratio[is.na(BostonHousing$ptratio)] <- mean(BostonHousing$ptratio, na.rm = T) # not run
summary(BostonHousing$ptratio)
Lets compute the accuracy when it is imputed with mean.
We use the function
regr.eval
from the DMwR package. This function compares the original vector with the modified one.
mape: mean absolute percentage error
"mae": mean absolute error, which is calculated as sum(|t_i - p_i|)/N, where t's are the true values and p's are the predictions, while N is supposed to be the size of both vectors.
"mse": mean squared error, which is calculated as sum( (t_i - p_i)^2 )/N
$$mae = \left ( \sum \left| (t_i - p_i) \right| \right) \frac {1}{N}$$ $$mse = \left ( \sum (t_i - p_i)^2 \right) \frac {1}{N}$$
$$rmse = \sqrt {mse}$$
$$mape = \left ( \sum \frac{{t_i - p_i}} {t_i} \right ) \frac {1}{N}$$
Here we will compare the 40 NA values replaced with the mean versus the original variable that had no NAs. It is only a 40values comparison of the variable ptratio
.
library(DMwR) BostonHousing <- original.NA summary(BostonHousing[c("rad", "ptratio")]) # summary before # get the 40 original non-NA values actuals <- original$ptratio[is.na(BostonHousing$ptratio)] # replace all the 40 NA values with the mean predicteds <- rep(mean(BostonHousing$ptratio, na.rm = TRUE), length(actuals)) # data.frame(actuals, predicteds) length(actuals); length(predicteds) # length of vectors to compare regr.eval(actuals, predicteds) # compare original vs modified variable #> mae mse rmse mape #> 1.62324034 4.19306071 2.04769644 0.095456
What is the Standard Error before an after imputation?
This is calculated over the vector ptratio
without any NAs.
BostonHousing <- original.NA # And here, we are replacing the 40 NAs in the whole `ptratio` vector. cat("Before: ", sum(is.na(BostonHousing$ptratio)), " NAs\n") summary(BostonHousing$ptratio) se.0 <- sd(BostonHousing$ptratio, na.rm = TRUE) / sqrt(sum(!is.na(BostonHousing$ptratio))) cat("SE(before) = ", se.0, "\n") indNA <- which(is.na(BostonHousing$ptratio)) # rows that are NA BostonHousing$ptratio[indNA] <- predicteds # replace NAs with the mean vector cat("\nAfter: ", sum(is.na(BostonHousing$ptratio)), " NAs\n") summary(BostonHousing$ptratio) se.1 <- sd(BostonHousing$ptratio, na.rm = TRUE) / sqrt(sum(!is.na(BostonHousing$ptratio))) cat("SE(after) = ", se.1, "\n") cat("\nImprovement = ", (se.0 - se.1) / se.0 * 100, " %")
Prediction is most advanced method to impute your missing values and includes different approaches such as: kNN Imputation, rpart, and mice.
DMwR::knnImputation uses k-Nearest Neighbours approach to impute missing values. What kNN imputation does in simpler terms is as follows: For every observation to be imputed, it identifies ‘k’ closest observations based on the euclidean distance and computes the weighted average (weighted based on distance) of these ‘k’ obs.
The advantage is that you could impute all the missing values in all variables with one call to the function. It takes the whole data frame as the argument and you don’t even have to specify which variable you want to impute. But be cautious not to include the response variable while imputing, because, when imputing in test/production environment, if your data contains missing values, you won’t be able to use the unknown response variable at that time.
library(DMwR) BostonHousing <- original.NA knnOutput <- knnImputation(BostonHousing[, !names(BostonHousing) %in% "medv"]) # perform knn imputation. anyNA(knnOutput) #> FALSE
Lets compute the accuracy.
actuals <- original$ptratio[is.na(BostonHousing$ptratio)] predicteds <- knnOutput[is.na(BostonHousing$ptratio), "ptratio"] regr.eval(actuals, predicteds) #> mae mse rmse mape #> 1.00188715 1.97910183 1.40680554 0.05859526
The mean absolute percentage error (mape) has improved by ~ 39% compared to the imputation by the mean. Good!
indNA <- which(is.na(BostonHousing$ptratio)) BostonHousing$ptratio[indNA] <- predicteds se.2 <- sd(BostonHousing$ptratio) / sqrt(sum(!is.na(BostonHousing$ptratio))) se.2
The improvement is r (se.0 - se.2) / se.0 * 100
percent
The limitation with DMwR::knnImputation is that it sometimes may not be appropriate to use when the missing value comes from a factor variable. Both rpart and mice has flexibility to handle that scenario. The advantage with rpart is that you just need only one of the variables to be non NA in the predictor fields.
The idea here is we are going to use rpart to predict the missing values instead of kNN. To handle factor variable, we can set the method=class while calling rpart(). For numeric, we use, method=anova. Here again, we need to make sure not to train rpart on response variable (medv).
library(rpart) BostonHousing <- original.NA class_mod <- rpart(rad ~ . - medv, data=BostonHousing[!is.na(BostonHousing$rad), ], method="class", na.action=na.omit) # since rad is a factor anova_mod <- rpart(ptratio ~ . - medv, data=BostonHousing[!is.na(BostonHousing$ptratio), ], method="anova", na.action=na.omit) # since ptratio is numeric. rad_pred <- predict(class_mod, BostonHousing[is.na(BostonHousing$rad), ]) ptratio_pred <- predict(anova_mod, BostonHousing[is.na(BostonHousing$ptratio), ])
Lets compute the accuracy for ptratio
actuals <- original$ptratio[is.na(BostonHousing$ptratio)] predicteds <- ptratio_pred regr.eval(actuals, predicteds) #> mae mse rmse mape #> 0.71061673 0.99693845 0.99846805 0.04099908
The mean absolute percentage error (mape) has improved additionally by another ~ 30% compared to the knnImputation. Very Good.
indNA <- which(is.na(BostonHousing$ptratio)) BostonHousing$ptratio[indNA] <- predicteds se.3 <- sd(BostonHousing$ptratio) / sqrt(sum(!is.na(BostonHousing$ptratio))) se.3
The improvement is r (se.0 - se.3) / se.0 * 100
percent.
rad
variableactuals <- original$rad[is.na(BostonHousing$rad)] predicteds <- as.numeric(colnames(rad_pred)[apply(rad_pred, 1, which.max)]) mean(actuals != predicteds) # compute misclass error. #> 0.25
mice short for Multivariate Imputation by Chained Equations is an R package that provides advanced features for missing value treatment. It uses a slightly uncommon way of implementing the imputation in 2-steps, using mice() to build the model and complete() to generate the completed data. The mice(df) function produces multiple complete copies of df, each with different imputations of the missing data. The complete() function returns one or several of these data sets, with the default being the first. Lets see how to impute ‘rad’ and ‘ptratio’:
library(mice) BostonHousing <- original.NA miceMod <- mice(BostonHousing[, !names(BostonHousing) %in% "medv"], method="rf") # perform mice imputation, based on random forests. miceOutput <- mice::complete(miceMod) # generate the completed data. anyNA(miceOutput) #> FALSE
Lets compute the accuracy of ptratio.
actuals <- original$ptratio[is.na(BostonHousing$ptratio)] predicteds <- miceOutput[is.na(BostonHousing$ptratio), "ptratio"] regr.eval(actuals, predicteds) #> mae mse rmse mape #> 0.36500000 0.78100000 0.88374204 0.02121326
The mean absolute percentage error (mape) has improved additionally by ~ 48% compared to the rpart. Excellent!.
indNA <- which(is.na(BostonHousing$ptratio)) BostonHousing$ptratio[indNA] <- predicteds anyNA(BostonHousing$ptratio) se.4 <- sd(BostonHousing$ptratio) / sqrt(sum(!is.na(BostonHousing$ptratio))) se.4 (se.0 - se.4) / se.0 * 100
Lets compute the accuracy of rad
actuals <- original$rad[is.na(BostonHousing$rad)] predicteds <- miceOutput[is.na(BostonHousing$rad), "rad"] mean(actuals != predicteds) # compute misclass error. #> 0.15
The mis-classification error reduced to 15%, which is 6 out of 40 observations. This is a good improvement compared to rpart’s 25%.
If you’d like to dig in deeper, here is the manual or in this other post about mice from DataScience+.
Though we have an idea of how each method performs, there is not enough evidence to conclude which method is better or worse. But these are definitely worth testing out the next time you impute missing values.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.