File: getTaskConstructorForLearner.R

package info (click to toggle)
r-cran-mlr 2.13-1
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 6,760 kB
  • sloc: ansic: 65; sh: 13; makefile: 2
file content (18 lines) | stat: -rw-r--r-- 513 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
 
getTaskConstructorForLearner = function(learner) {
  while (inherits(learner, "BaseWrapper"))
    learner = learner$next.learner
  cl = class(learner)

  if ("RLearnerRegr" %in% cl) {
    constructor = makeRegrTask
  } else if ("RLearnerClassif" %in% cl) {
    constructor = makeClassifTask
  } else if ("RLearnerSurv" %in% cl) {
    constructor = makeSurvTask
  } else if ("RLearnerCluster" %in% cl) {
    constructor = makeClusterTask
  } else {
    stop("Unknown learner class for impute")
  }
  constructor
}