R/missingValueImputation.R

In urmanml/easyMLR: Easy Machine Learning - end to end ML worklflow in simple english language

#' Function for missing value imputation
#' missing value imputation based on the class of variable


missingValueImputation<-function(dataset,target)
{
  imp = impute(dataset,target = target, classes = list(integer = imputeMean(),numeric=imputeMean(), factor = imputeMode()),
               dummy.classes = c("integer","numeric"))

  dataset<-imp$data
  return(dataset)
}