kapsner / mlsurvlrnrs

Survival learners for the `mlexperiments` R 📦

mlsurvlrnrs

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.

Installation

To install the development version, run

install.packages("remotes")
remotes::install_github("kapsner/mlsurvlrnrs")