Skip to contents

Routes to one of three SHAP views. type = "beeswarm" (default) draws the signature SHAP summary; "importance" draws a mean-absolute-SHAP bar chart; "dependence" draws SHAP value against a single feature's value.

Usage

# S3 method for class 'gg_shap'
plot(
  x,
  type = c("beeswarm", "importance", "dependence"),
  xvar = NULL,
  labels = NULL,
  ...
)

Arguments

x

A gg_shap object.

type

One of "beeswarm", "importance", or "dependence".

xvar

For type = "dependence", the variable to plot. When NULL, the top-ranked variable is used.

labels

Optional variable labels, forwarded to the mode function. One of: a named character vector (c(Temp = "Temperature")); a labelled data frame, whose attr(col, "label") values are read; or a two-column key/label data frame. Variables with no label keep their raw name. In "dependence" mode the label is substituted into the axis titles; xvar still matches on raw names.

...

Passed to the underlying builder.

Value

A ggplot object.

Examples

# \donttest{
if (requireNamespace("kernelshap", quietly = TRUE)) {
  rf <- randomForestSRC::rfsrc(Ozone ~ ., data = na.omit(airquality),
                               ntree = 50)
  gg_dta <- gg_shap(rf, bg_n = 20)
  plot(gg_dta, type = "importance")
}

# }