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")