Table 5 Search space of the hyper-parameters explored for each model.

From: Alzheimer’s disease risk prediction using machine learning for survival analysis with a comorbidity-based approach

Model

Parameters

Search Space

Random Survival Forest

number of trees

1 to 1000, logscale=TRUE*

node depth

1 to 100

number of variables considered at each split (mtry)

1 to 36

split rule

{logrank, logrankscore}

Fast Random Survival Forest

min node size

1 to 100

number of variables considered at each split (mtry)

1 to 36

number of trees

1 to 1000, logscale=TRUE*

Cvglmnet

alpha

(0,1)

lambda.min.ratio

(0,1)

Rpart

complexity parameter

(0,1)

DeepSurv

dropout

(0,1)

learning rate

(0,1)

DeepHit

dropout

(0,1)

learning rate

(0,1)

CoxTime

dropout

(0,1)

learning rate

(0,1)

  1. * This refers to sampling the hyperparameter from a logarithmic scale rather than a linear one. This improves efficiency for large search spaces by favoring smaller values while still exploring larger ones, as supported by the mlr3 framework.