This Learner specializes mlr3::Learner for survival problems. Predefined learners can be found in the mlr3misc::Dictionary mlr3::mlr_learners.

Format

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

Construction

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". Possible values for predict_type are "distr", "lp", and "crank".

Fields

See mlr3::Learner.

Methods

See mlr3::Learner.

Examples

library(mlr3) ids = mlr_learners$keys("^surv") ids
#> [1] "surv.blackboost" "surv.coxph" "surv.cvglmnet" #> [4] "surv.flexible" "surv.gamboost" "surv.gbm" #> [7] "surv.glmboost" "surv.glmnet" "surv.kaplan" #> [10] "surv.mboost" "surv.nelson" "surv.parametric" #> [13] "surv.penalized" "surv.randomForestSRC" "surv.ranger" #> [16] "surv.rpart" "surv.svm"
# get a specific learner from mlr_learners: lrn = mlr_learners$get("surv.rpart") print(lrn)
#> <LearnerSurvRpart:surv.rpart> #> * Model: - #> * Parameters: xval=0 #> * Packages: rpart, distr6, survival #> * Predict Type: crank #> * Feature types: logical, integer, numeric, character, factor, ordered #> * Properties: importance, missings, selected_features, weights