plot.surv_shap function

Plot SurvSHAP(t) Explanations for Survival Models

Plot SurvSHAP(t) Explanations for Survival Models

This functions plots objects of class surv_shap - time-dependent explanations of survival models created using the predict_parts(..., type="survshap") function.

## S3 method for class 'surv_shap' plot( x, ..., title = "SurvSHAP(t)", subtitle = "default", max_vars = 7, colors = NULL, rug = "all", rug_colors = c("#dd0000", "#222222") )

Arguments

  • x: an object of class surv_shap to be plotted
  • ...: additional objects of class surv_shap to be plotted together
  • title: character, title of the plot
  • subtitle: character, subtitle of the plot, 'default' automatically generates "created for XXX, YYY models", where XXX and YYY are the explainer labels
  • max_vars: maximum number of variables to be plotted (least important variables are ignored)
  • colors: character vector containing the colors to be used for plotting variables (containing either hex codes "#FF69B4", or names "blue")
  • rug: character, one of "all", "events", "censors", "none" or NULL. Which times to mark on the x axis in geom_rug().
  • rug_colors: character vector containing two colors (containing either hex codes "#FF69B4", or names "blue"). The first color (red by default) will be used to mark event times, whereas the second (grey by default) will be used to mark censor times.

Returns

An object of the class ggplot.

Examples

library(survival) library(survex) model <- randomForestSRC::rfsrc(Surv(time, status) ~ ., data = veteran) exp <- explain(model) p_parts_shap <- predict_parts(exp, veteran[1, -c(3, 4)], type = "survshap") plot(p_parts_shap)

See Also

Other functions for plotting 'predict_parts_survival' objects: plot.predict_parts_survival(), plot.surv_lime()

  • Maintainer: Mikołaj Spytek
  • License: GPL (>= 3)
  • Last published: 2023-10-24