R/RemoveConstantFeaturesWrapper.R

In shuodata/mlr-master: Machine Learning in R

#' @title Fuse learner with removal of constant features preprocessing.
#'
#' @description
#' Fuses a base learner with the preprocessing implemented in \code{\link{removeConstantFeatures}}.
#'
#' @template arg_learner
#' @inheritParams removeConstantFeatures
#' @export
#' @family wrapper
#' @template ret_learner
makeRemoveConstantFeaturesWrapper = function(learner, perc = 0, dont.rm = character(0L), na.ignore = FALSE, tol = .Machine$double.eps^.5) {
  learner = checkLearner(learner)
  args = list(perc = perc, dont.rm = dont.rm, na.ignore = na.ignore, tol = tol)
  rm(list = names(args))

  trainfun = function(data, target, args) {
    args$dont.rm = union(args$dont.rm, target)
    tmp = do.call(removeConstantFeatures, c(list(obj = data), args))
    list(data = tmp, control = list(dropped.cols = setdiff(names(data), names(tmp))))
  }

  predictfun = function(data, target, args, control) {
    dropNamed(data, control$dropped.cols)
  }

  lrn = makePreprocWrapper(learner, trainfun, predictfun, par.vals = args)
  addClasses(lrn, "RemoveConstantFeaturesWrapper")
}