File: SupervisedTask.R

package info (click to toggle)
r-cran-mlr 2.19.2%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 8,264 kB
  • sloc: ansic: 65; sh: 13; makefile: 5
file content (38 lines) | stat: -rw-r--r-- 1,329 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
makeSupervisedTask = function(type, data, target, weights, blocking, fixup.data, check.data, coordinates) {
  task = makeTask(type = type, data = data, weights = weights, blocking = blocking, fixup.data = fixup.data, check.data = check.data, coordinates = coordinates)

  if (check.data) {
    # costsens does not have a target col...
    # assertCharacter(target, any.missing = FALSE, min.len = 1L)
    w = which.first(target %nin% colnames(data))
    if (length(w) > 0L) {
      stopf("Column names of data doesn't contain target var: %s", target[w])
    }
    checkTaskData(task$env$data, cols = setdiff(colnames(data), target))
  }

  addClasses(task, "SupervisedTask")
}

#' @export
print.SupervisedTask = function(x, print.target = TRUE, print.weights = TRUE, ...) {

  td = x$task.desc
  catf("Supervised task: %s", td$id)
  catf("Type: %s", td$type)
  if (print.target) {
    catf("Target: %s", collapse(td$target))
  }
  if (inherits(x, "SurvTask")) {
    catf("Events: %i", sum(getTaskTargets(x)[, 2L]))
  }
  catf("Observations: %i", td$size)
  catf("Features:")
  catf(printToChar(td$n.feat, collapse = "\n"))
  catf("Missings: %s", td$has.missings)
  if (print.weights) {
    catf("Has weights: %s", td$has.weights)
  }
  catf("Has blocking: %s", td$has.blocking)
  catf("Has coordinates: %s", td$has.coordinates)
}