in R/step1_data_preprocessing.R [228:243]
fill_NA_mode_mean <- function(data){
data <- data.frame(data)
for(j in 1:length(categ)){
row_na <- which(is.na(data[,categ[j]]) == TRUE)
if(length(row_na > 0)){
data[row_na, categ[j]] <- subset(Mode, name == categ[j])[1,2]
}
}
for(j in 1:length(contin)){
row_na <- which(is.na(data[,contin[j]]) == TRUE)
if(length(row_na > 0)){
data[row_na, contin[j]] <- subset(Mean, name == contin[j])[1,2]
}
}
return(data)
}