TaskSurv: Survival Task

In mlr-org/mlr3survival: Survival Analysis for mlr3

Description

This task specializes mlr3::Task and mlr3::TaskSupervised for right-censored survival problems. The target column is assumed to be a factor. Predefined tasks are stored in mlr3::mlr_tasks.

The task_type is set to "surv".

Format

R6::R6Class object inheriting from Task/TaskSupervised.

Construction

t = TaskSurv$new(id, backend, time, status)

• id :: character(1)
  Name of the task.

• backend :: DataBackend

• time :: numeric()
  Event times.

• status :: integer() | logical()
  Event indicator. "0"/FALSE means alive (no event), "1"/TRUE means dead (event).

Fields

See mlr3::TaskSupervised.

Methods

All methods from mlr3::TaskSupervised, and additionally:

• survfit(strata = character())
character() -> survival::survfit()
  Creates a survival::survfit() object for the survival times. Argument strata can be used to stratify into multiple groups.

Examples

library(mlr3)
lung = mlr3misc::load_dataset("lung", package = "survival")
lung$status = (lung$status == 2L)
b = as_data_backend(lung)
task = TaskSurv$new("lung", backend = b, time = "time", status = "status")

task$target_names
task$feature_names
task$formula()
task$truth()
task$survfit("age > 50")

mlr-org/mlr3survival documentation built on Oct. 21, 2019, 7:42 p.m.