Plot the results of a survival glmnet model fit on sample-level data.

Plot the results of a survival glmnet model fit on sample-level data.

Usage

tof_plot_model_survival(
  tof_model,
  new_data,
  censor_size = 2.5,
  theme = ggplot2::theme_bw()
)

Arguments

tof_model

A `tof_model` trained using tof_train_model

new_data

A tibble of new observations for which a plot should be made. If new_data isn't provided, the plot will be made using the training data used to fit the model. Alternatively, the string "tuning_data" can be provided, and the plot will be generated using the predictions generated during model tuning.

censor_size

A numeric value indicating how large to plot the tick marks representing censored values in the Kaplan-Meier curve.

theme

A ggplot2 theme to apply to the plot. Defaults to theme_bw

Value

A ggplot object. Specifically, a Kaplan-Meier curve.