The goal of mlsurvlrnrs is to enhance the mlexperiments R package with survival learners.
Currently implemented learners are:
| Name | Based on | Description / Tasks |
|---|---|---|
| LearnerSurvCoxPHCox | survival::coxph |
Cox Proportional Hazards Regression |
| LearnerSurvGlmnetCox | glmnet::glmnet |
Regularized Cox Regression |
| LearnerSurvRangerCox | ranger::ranger |
Random Survival Forest with right-censored data |
| LearnerSurvRpartCox | rpart::rpart |
Random Survival Forest with right-censored data |
| LearnerSurvXgboostCox | xgboost::xgb.train |
Cox Regression with right-censored data |
| LearnerSurvXgboostAft | xgboost::xgb.train |
Accelerated failure time models with right-censored data |
For a short introduction on how to use the learners together with the mlexperiments R package, please visit the wiki page.
To install the development version, run
install.packages("remotes")
remotes::install_github("kapsner/mlsurvlrnrs")