R/TaskClassif_pima.R

In mllg/mlr3: Machine Learning in R - Next Generation

#' @title Pima Indian Diabetes Classification Task
#'
#' @name mlr_tasks_pima
#' @format [R6::R6Class] inheriting from [TaskClassif].
#' @include mlr_tasks.R
#'
#' @section Construction:
#' ```
#' mlr_tasks$get("pima")
#' tsk("pima")
#' ```
#'
#' @description
#' A classification task for the [mlbench::PimaIndiansDiabetes2] data set.
#' Positive class is set to `"pos"`.
#'
#' @template seealso_task
NULL

load_task_pima = function(id = "pima") {
  b = as_data_backend(load_dataset("PimaIndiansDiabetes2", "mlbench"))
  b$hash = "_mlr3_tasks_pima_"
  TaskClassif$new(id, b, target = "diabetes", positive = "pos")
}

#' @include mlr_tasks.R
mlr_tasks$add("pima", load_task_pima)