R/getLearner.R

In rgmantovani/HpTuning: An implementation of several hyperparameter tuning techniques for ML classification algorithms in R.

# -------------------------------------------------------------------------------------------------
# -------------------------------------------------------------------------------------------------

# [*] mlr Wrapper learners
# http://mlr-org.github.io/mlr-tutorial/release/html/wrapper/index.html

# -------------------------------------------------------------------------------------------------
# -------------------------------------------------------------------------------------------------

getLearner = function(algo) {

  if(algo == "classif.glmnet") {
    lrn = makeLearner(algo, alpha = 0)
  } else {
    lrn = makeLearner(algo)
  }

  lrn = makeImputeWrapper(
    learner = lrn,
    classes = list(numeric = imputeMedian(), factor = imputeMode(), integer = imputeMedian()),
    dummy.classes = c("numeric", "factor", "integer")
  )

  new.lrn = makeRemoveConstantFeaturesWrapper(learner = lrn)
  return(new.lrn)
}

# -------------------------------------------------------------------------------------------------
# -------------------------------------------------------------------------------------------------