survival learners R

surv_learners.R

# Create learners----
learners <- lrns(
  paste0("surv.", c("coxtime","deephit", "deepsurv", "loghaz", "pchazard")),
  frac = 0.3, activation = "relu",
  dropout = 0.1, 
  early_stopping = TRUE, 
  epochs = 10,
  batch_size = 32L
  )
learners <- c(learners, lrns(c("surv.kaplan", "surv.coxph")))

# Create pipeline----
pipeline <- function(learner){
  po("scale") %>>% po("learner", learner)
}
learners <- lapply(learners, pipeline)