plot.predict_profile_survival {survex} | R Documentation |
Plot Predict Profile for Survival Models
Description
This function plots objects of class "predict_profile_survival"
created using
the predict_profile()
function.
Usage
## S3 method for class 'predict_profile_survival'
plot(
x,
...,
geom = "time",
variables = NULL,
variable_type = NULL,
facet_ncol = NULL,
numerical_plot_type = "lines",
times = NULL,
marginalize_over_time = FALSE,
title = "default",
subtitle = "default",
colors = NULL,
rug = "all",
rug_colors = c("#dd0000", "#222222")
)
Arguments
x |
an object of class |
... |
additional objects of class |
geom |
character, either |
variables |
character, names of the variables to be plotted. When |
variable_type |
character, either |
facet_ncol |
number of columns for arranging subplots. Only used when |
numerical_plot_type |
character, either |
times |
numeric vector, times for which the profile should be plotted, the times must be present in the 'times' field of the explainer. If |
marginalize_over_time |
logical, if |
title |
character, title of the plot |
subtitle |
character, subtitle of the plot, |
colors |
character vector containing the colors to be used for plotting variables (containing either hex codes "#FF69B4", or names "blue") |
rug |
character, one of |
rug_colors |
character vector containing two colors (containing either hex codes |
Value
A collection of ggplot
objects arranged with the patchwork
package.
Examples
library(survival)
library(survex)
model <- randomForestSRC::rfsrc(Surv(time, status) ~ ., data = veteran)
exp <- explain(model)
p_profile <- predict_profile(exp, veteran[1, -c(3, 4)])
plot(p_profile)
p_profile_with_cat <- predict_profile(
exp,
veteran[1, -c(3, 4)],
categorical_variables = c("trt", "prior")
)
plot(p_profile_with_cat)