Skip to content

Instantly share code, notes, and snippets.

@RaphaelS1
Last active April 18, 2021 08:20
Show Gist options
  • Save RaphaelS1/95e419bf2853dbf731ae0cbce243aab3 to your computer and use it in GitHub Desktop.
Save RaphaelS1/95e419bf2853dbf731ae0cbce243aab3 to your computer and use it in GitHub Desktop.
## learners are stored in mlr3extralearners
library(mlr3extralearners)
## load learners
learners <- lrns(
 paste0("surv.", c("coxtime", "deephit", "deepsurv", "loghaz", "pchazard")),
 frac = 0.3, early_stopping = TRUE, epochs = 10, optimizer = "adam"
)
# apply our function
learners <- lapply(learners, create_autotuner)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment