Skip to contents

Aalen Johansen Competing Risks Learner

Source: R/LearnerCompRisksAalenJohansen.R
mlr_learners_cmprsk.aalen.Rd

This learner estimates the Cumulative Incidence Function (CIF) for competing risks using the empirical Aalen-Johansen (AJ) estimator.

Transition probabilities to each competing event are computed from the training data via the survival::survfit.formula() function. Predictions are made at all unique event times (across all causes) observed in the training set.

Dictionary

This Learner can be instantiated via the dictionary mlr_learners or with the associated sugar function lrn():

mlr_learners$get("cmprsk.aalen")
lrn("cmprsk.aalen")

Meta Information

  • Task type: “cmprsk”

  • Predict Types: “cif”

  • Feature Types: “logical”, “integer”, “numeric”, “factor”

  • Required Packages: mlr3, mlr3cmprsk, survival

Parameters

Empty ParamSet

References

Aalen, O O, Johansen, Soren (1978). “An empirical transition matrix for non-homogeneous Markov chains based on censored observations.” Scandinavian journal of statistics, 141–150.

See also

Other competing risk learners: mlr_learners_cmprsk.fg

Super classes

mlr3::Learner -> LearnerCompRisks -> LearnerCompRisksAalenJohansen

Active bindings

native_model

(survival::survfit)
The fitted model.

Methods

Public methods

Inherited methods

LearnerCompRisksAalenJohansen$new()

Creates a new instance of this R6 class.

Usage

LearnerCompRisksAalenJohansen$new()

LearnerCompRisksAalenJohansen$importance()

All features have a score of 0 for this learner. This method exists solely for compatibility with the mlr3 ecosystem, as this learner is used as a fallback for other survival learners that require an importance() method.

Usage

LearnerCompRisksAalenJohansen$importance()

Returns

Named numeric().

LearnerCompRisksAalenJohansen$selected_features()

Selected features are always the empty set for this learner. This method is implemented only for compatibility with the mlr3 API, as this learner does not perform feature selection.

Usage

LearnerCompRisksAalenJohansen$selected_features()

Returns

character(0).

LearnerCompRisksAalenJohansen$clone()

The objects of this class are cloneable with this method.

Usage

LearnerCompRisksAalenJohansen$clone(deep = FALSE)

Arguments

deep

Whether to make a deep clone.

Examples

# Define a task
task = tsk("pbc")
task$select(c("age", "chol", "albumin", "ast", "bili", "protime"))

# Stratify by event status
task$set_col_roles(cols = "status", add_to = "stratum")
task
#> 
#> ── <TaskCompRisks> (276x8): Primary Biliary Cholangitis ────────────────────────
#> • Target: time and status
#> • Properties: strata
#> • Features (6):
#>   • dbl (4): albumin, ast, bili, protime
#>   • int (2): age, chol
#> • Strata: status

# Create training and test sets
set.seed(42L)
part = partition(task)

# Define the learner
learner = lrn("cmprsk.aalen")

# Train on the training set
learner$train(task, row_ids = part$train)
learner$native_model
#> Call: survfit(formula = task$formula(1), data = task$data(cols = task$target_names), 
#>     weights = NULL)
#> 
#>        n nevent     rmean se(rmean)*
#> (s0) 184      0 90.653492   4.062322
#> 1    184     12  7.979854   2.225116
#> 2    184     74 50.366654   4.044563
#>    *restricted mean time in state (max time = 149 )

# Predict CIFs for the test set
predictions = learner$predict(task, row_ids = part$test)
predictions
#> 
#> ── <PredictionCompRisks> for 92 observations: ──────────────────────────────────
#>  row_ids time event       CIF
#>        1   13     2 <list[2]>
#>       10  123     2 <list[2]>
#>       16    4     2 <list[2]>
#>      ---  ---   ---       ---
#>      157   73     1 <list[2]>
#>      208   27     1 <list[2]>
#>      213   47     1 <list[2]>

# Score predictions
predictions$score()
#> cmprsk.ibs 
#>  0.1780101