h2o.shap.varimp_plot: h2o.shap.varimp_plot
In 6chaoran/shapvis: functions for shap values visualization

Description Usage Arguments Examples

View source: R/h2o_shap.R

h2o shap variable importance plot

1	h2o.shap.varimp_plot(shap_values, X, max.display = 20, plot_type = "bar")

`shap_values`	shap value matrix from `h2o.predict_contribution`
`X`	original dataset
`max.display`	number of variables to show in the plot
`plot_type`	'dot','bar'

data(shap_values_mtcars)
data(mtcars)
h2o.shap.varimp_plot(shap_values_mtcars, mtcars, 20, 'bar')
h2o.shap.varimp_plot(shap_values_mtcars, mtcars, 20, 'dot')