todo-files/RLearner_regr_bagEarth.R

In mlr-org/mlr: Machine Learning in R

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

trainLearner.regr.bagEarth = function(.learner, .task, .subset, ...) {
  f = getTaskFormula(.task)
  d = getTaskData(.task, .subset)
  if (.task$task.desc$has.weights) {
    f = getTaskFormula(.task)
    bagEarth(f, data = getTaskData(.task, .subset), ...)
  } else {
    bagEarth(f, data = d, ...)
  }
}

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