todo-files/RLearner_regr_bagEarth.R

In Najah-lshanableh/R-data-mining2: 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)
}