h2o.shap.waterfall_plot: h2o.shap.waterfall_plot

In 6chaoran/shapvis: functions for shap values visualization

Description

h2o shap waterfall plot

Usage

h2o.shap.waterfall_plot(
  shap_values,
  X,
  idx,
  link = "",
  centre = FALSE,
  max.display = 20,
  type = "bar"
)

Arguments

shap_values	shap value matrix from h2o.predict_contribution
X	original dataset
idx	index of the individual prediction
link	linking function ”, 'logit'. default NULL: identical, alternative logit: logit transform to probability
centre	whether shap.value centred at 0, probabilty centred at 0.5, default FALSE FALSE: add bias term back
max.display	number of variables to show in the plot
type	bar, arrow

Value

ggplot2 waterfall plot

Examples

data(mtcars)
data(shap_values_mtcars)
h2o.shap.waterfall_plot(shap_values_mtcars, X = mtcars, idx = 1)

6chaoran/shapvis documentation built on July 29, 2020, 5:31 p.m.