Learn R Programming

SHAPforxgboost (version 0.1.3)

shap.importance: Variable importance as measured by mean absolute SHAP value.

Description

Variable importance as measured by mean absolute SHAP value.

Usage

shap.importance(data_long, names_only = FALSE, top_n = Inf)

Value

returns data.table with average absolute SHAP values per variable, sorted in decreasing order of importance.

Arguments

data_long

a long format data of SHAP values from shap.prep

names_only

If TRUE, returns variable names only.

top_n

How many variables to be returned?

Examples

Run this code

shap.importance(shap_long_iris)

shap.importance(shap_long_iris, names_only = 1)

Run the code above in your browser using DataLab