CRAN Status
Badge CRAN
Checks tic CodeFactor

Status Lifecycle

Downloads codecov dependencies License:

What is survivalmodels?

survivalmodels implements models for survival analysis that are either not already implemented in R, or novel implementations for speed improvements. Currently implemented are five neural networks from the Python packages pycox, DNNSurv, and the Akritas non-parametric conditional estimator. Further updates will include implementations of novel survival models.

For a hands-on demonstration of model training, tuning, and comparison see this article I wrote, which uses the mlr3proba interface with models from survivalmodels.

Basic Usage

# load dependencies

train <- simsurvdata(100)
test <- simsurvdata(50)

fit <- akritas(Surv(time, status) ~ ., data = train)
predict(fit, newdata = test)

# Use distr6 = TRUE to return a distribution
predict_distr <- predict(fit, newdata = test, distr6 = TRUE)

# Return a relative risk ranking with type = "risk"
predict(fit, newdata = test, type = "risk")

Or both survival probabilities and a rank
predict(fit, newdata = test, type = "all", distr6 = TRUE)

Python Models

survivalmodels implements models from Python using reticulate. In order to use these models, the required Python packages must be installed following with reticulate::py_install. survivalmodels includes a helper function to install the required pycox function (with pytorch if also required). Before running any models in this package, if you have not already installed pycox please run

install_pycox(pip = TRUE, install_torch = FALSE)

With the arguments changed as you require, see ?install_pycox for more.

For DNNSurv the model depends on keras and tensorflow, which require installation via:

install_keras(pip = TRUE, install_tensorflow = FALSE)


Install the latest release from CRAN:


Install the development version from GitHub:


Try the survivalmodels package in your browser

Any scripts or data that you put into this service are public.

survivalmodels documentation built on March 24, 2022, 9:05 a.m.