todo-files/RLearner_regr_icr.R

In mlr-org/mlr: Machine Learning in R

makeRLearner.regr.icr = function() {
  makeRLearnerRegr(
    cl = "regr.icr",
    package = "stats",
    missings = FALSE,
    numerics = TRUE,
    factors = TRUE,
    se = FALSE,
    weights = FALSE
  )
}

trainLearner.regr.icr = function(.learner, .task, .subset, ...) {
  f = getTaskFormula(.task)
  icr(f, data = getTaskData(.task, .subset), ...)
}

predictLearner.regr.icr = function(.learner, .model, .newdata, ...) {
  predict(.model$learner.model, newdata = .newdata)
}