todo-files/RLearner_regr_icr.R

In guillermozbta/mir: 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)
}