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
|
#' @title Create a regression task.
#' @inheritParams Task
#' @seealso [Task] [ClassifTask] [CostSensTask] [ClusterTask] [MultilabelTask] [SurvTask]
#' @rdname RegrTask
#' @aliases RegrTask
#' @export
makeRegrTask = function(id = deparse(substitute(data)), data, target, weights = NULL, blocking = NULL, coordinates = NULL, fixup.data = "warn", check.data = TRUE) {
assertString(id)
assertDataFrame(data)
assertString(target)
assertChoice(fixup.data, choices = c("no", "quiet", "warn"))
assertFlag(check.data)
if (fixup.data != "no") {
if (is.integer(data[[target]])) {
data[[target]] = as.double(data[[target]])
}
}
task = makeSupervisedTask("regr", data, target, weights, blocking, coordinates, fixup.data = fixup.data, check.data = check.data)
if (check.data) {
assertNumeric(data[[target]], any.missing = FALSE, finite = TRUE, .var.name = target)
}
task$task.desc = makeRegrTaskDesc(id, data, target, weights, blocking, coordinates)
addClasses(task, "RegrTask")
}
#' @export
#' @rdname makeTaskDesc
makeRegrTaskDesc = function(id, data, target, weights, blocking, coordinates) {
addClasses(makeTaskDescInternal("regr", id, data, target, weights, blocking, coordinates), c("RegrTaskDesc", "SupervisedTaskDesc"))
}
|
