plot.model_parts_survival: Plot Model Parts for Survival Models

In survex: Explainable Machine Learning in Survival Analysis

Plot Model Parts for Survival Models

Description

This function is a wrapper for plotting model_parts objects created for survival models using the model_parts() function.

Usage

## S3 method for class 'model_parts_survival'
plot(x, ...)

Arguments

x	an object of class "model_parts_survival" to be plotted
...	additional parameters passed to the plot.surv_feature_importance function

Value

An object of the class ggplot.

Plot options

• title - character, title of the plot

• subtitle - character, subtitle of the plot, if NULL automatically generated as "created for XXX, YYY models", where XXX and YYY are 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.

See Also

Other functions for plotting 'model_parts_survival' objects: plot.surv_feature_importance()

Examples

library(survival)
library(survex)

model <- coxph(Surv(time, status) ~ ., data = veteran, x = TRUE, model = TRUE, y = TRUE)
explainer <- explain(model)

mp <- model_parts(explainer)

plot(mp)

survex documentation built on Oct. 25, 2023, 1:06 a.m.