R/getLearners.R

In rgmantovani/autoHyperBot:

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

getAllPossibleLearners = function(filled.task) {

  obj = convertOMLTaskToMlr(filled.task)
  learners.list = listLearners(obj$mlr.task, create = TRUE, properties = "prob")

  return(learners.list)
}

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

getPredefinedLearners = function() {

  learners.list = lapply(predefined.learners, makeLearner, predict.type = "prob")
  return(learners.list)
}

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