vip2vivid | R Documentation |
Takes measured importance and interactions from the vip package and turns them into a matrix which can be used for plotting. Accepts any of the variable importance methods supplied by vip.
vip2vivid(importance, interaction, reorder = TRUE)
importance |
Measured importance from the vip package using |
interaction |
Measured interaction from the vip package using |
reorder |
If TRUE (default) uses DendSer to reorder the matrix of interactions and variable importances. |
A matrix of interaction values, with importance on the diagonal.
## Not run:
library(ranger)
library(vip)
aq <- na.omit(airquality) # get data
nameAq <- names(aq[-1]) # get feature names
rF <- ranger(Ozone ~ ., data = aq, importance = "permutation") # create ranger random forest fit
vImp <- vi(rF) # vip importance
vInt <- vint(rF, feature_names = nameAq) # vip interaction
vip2vivid(vImp, vInt)
## End(Not run)
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.