addLearner: Add Learners to 'LearnerList'

Description Usage Arguments Details See Also Examples

View source: R/learner.R

Description

The function can be used to add new learner to LearnerList in the current R session.

Usage

1

Arguments

x

a list containing all required information to define a new learner (see Details below).

Details

The function can be used to add new learners to LearnerList in the current R session. The function expects a list of four elements including the name of the learners object class, the name of the package where the class and the fitting method is implemented, the name of the method and a prediction function that predicts class probabilities (in the classification case) or numeric values (in the regression case) and takes the arguments x (the fitted model object), newdata a data.frame containing the predictions of the observations in the evaluation sample and yclass a character string specifying the type of the response variable ("numeric", "factor", etc.). The elements in the list should be named class, package, method and predfun.

See Also

LearnerList

Examples

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
newlearner <- list(
  class   = "svm",
  package = "e1071",
  method  = "Support Vector Machine",
  predict = function(x, newdata, yclass = NULL) {
    if(match(yclass, c("ordered", "factor"))) {
      attr(predict(x, newdata = newdata, probability = TRUE), "probabilities")
    } else {
      predict(x, newdata = newdata)
    }
  })

addLearner(newlearner)

stablelearner documentation built on Jan. 13, 2021, 5:41 p.m.