LearnerSurv: Survival Learner

Description Format Construction Fields Methods Examples

Description

This Learner specializes mlr3::Learner for survival problems. The slot task_type is set to "surv". Predefined learners can be found in the mlr3misc::Dictionary mlr3::mlr_learners.

Format

R6::R6Class object inheriting from mlr3::Learner.

Construction

1
2
l = LearnerSurv$new(id, param_set = ParamSet$new(), predict_types = character(),
     feature_types = character(), properties = character(), packages = character())

For a description of the arguments, see mlr3::Learner. task_type is set to "surv". Currently only possible value for predict_types is "risk".

Fields

See mlr3::Learner.

Methods

See mlr3::Learner.

Examples

1
2
3
4
5
6
7
8
library(mlr3)
ids = mlr_learners$keys("^surv")
lrns = mlr_learners$mget(ids)
names(lrns)

# get a specific learner from mlr_learners:
lrn = mlr_learners$get("surv.rpart")
print(lrn)

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