Revision 272ac623c984dbf5defce76b6495c05accafe79f authored by Patrick Schratz on 17 December 2019, 04:08:28 UTC, committed by Patrick Schratz on 17 December 2019, 04:08:28 UTC
Build URL: https://circleci.com/gh/mlr-org/mlr/1264 Commit:
1 parent 9de9c6e
MultilabelTask.R
#' @title Create a multilabel task.
#' @inheritParams Task
#' @section Note:
#' For multilabel classification we assume that the presence of labels is encoded via logical
#' columns in `data`. The name of the column specifies the name of the label. `target`
#' is then a char vector that points to these columns.
#' @seealso [Task] [ClassifTask] [ClusterTask] [CostSensTask] [RegrTask] [SurvTask]
#' @details
#' For multilabel classification we assume that the presence of labels is encoded via logical
#' columns in `data`. The name of the column specifies the name of the label. `target`
#' is then a char vector that points to these columns.
#' @rdname MultilabelTask
#' @aliases MultilabelTask
#' @export
makeMultilabelTask = function(id = deparse(substitute(data)), data, target, weights = NULL,
blocking = NULL, coordinates = NULL, fixup.data = "warn", check.data = TRUE) {
assertString(id)
assertCharacter(target, any.missing = FALSE, min.len = 2L)
assertDataFrame(data)
assertChoice(fixup.data, choices = c("no", "quiet", "warn"))
assertFlag(check.data)
task = makeSupervisedTask("multilabel", data = data, target = target,
weights = weights, blocking = blocking,
coordinates = coordinates, fixup.data = fixup.data,
check.data = check.data)
# currently we dont do any fixup here
if (check.data) {
for (cn in target) {
assertLogical(task$env$data[[cn]], any.missing = FALSE, .var.name = cn)
}
}
task$task.desc = makeMultilabelTaskDesc(id, data, target, weights, blocking, coordinates)
addClasses(task, "MultilabelTask")
}
#' @export
print.MultilabelTask = function(x, ...) {
y = getTaskTargets(x)
sums = colSums(y)
print.SupervisedTask(x)
catf("Classes: %i", ncol(y))
print(sums)
}
#' @export
#' @rdname makeTaskDesc
makeMultilabelTaskDesc = function(id, data, target, weights, blocking, coordinates) {
levs = target
td = makeTaskDescInternal("multilabel", id, data, target, weights, blocking, coordinates)
td$class.levels = levs
return(addClasses(td, c("MultilabelTaskDesc", "SupervisedTaskDesc")))
}
Computing file changes ...