'See vip for each class in multi-class randomforest?
In this example of multi-class classification using randomforest model, the author creates a vip chart as shown below. Is there a way to view which variables are influencing the model to predict each of the response categories? For example, I'd like to be able to see that "x" variable is driving something to be classified as "x" volcano type (response category).
https://juliasilge.com/blog/multinomial-volcano-eruptions/
library(vip)
rf_spec %>%
set_engine("ranger", importance = "permutation") %>%
fit(
volcano_type ~ .,
data = juice(volcano_prep) %>%
select(-volcano_number) %>%
janitor::clean_names()
) %>%
vip(geom = "point")
Solution 1:[1]
Created an explainer using explain_tidymodels, then used the DALEX vignette.
https://www.tmwr.org/explain.html https://modeloriented.github.io/DALEX/articles/multilabel_classification.html
Sources
This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.
Source: Stack Overflow
| Solution | Source |
|---|---|
| Solution 1 | ava |
