powered by
This implies to use colnames(x) to get the column names of the SHAP and feature matrix (and optional SHAP interaction values).
colnames(x)
# S3 method for shapviz dimnames(x)
Dimnames of the SHAP matrix.
An object of class "shapviz".
shapviz()
S <- matrix(c(1, -1, -1, 1), ncol = 2, dimnames = list(NULL, c("x", "y"))) X <- data.frame(x = c("a", "b"), y = c(100, 10)) x <- shapviz(S, X, baseline = 4) dimnames(x) colnames(x)
Run the code above in your browser using DataLab