You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I'm not sure adding it as an option to the function is a general use case. However since plot_predict_interaction returns a ggplot object, you can update the labels on top of it:
library(randomForestExplainer)
library(randomForest)
library(ggplot2)
data(Boston, package = "MASS")
Boston$chas <- as.logical(Boston$chas)
set.seed(2017)
forest <- randomForest(medv ~ ., data = Boston, localImp = TRUE)
p <- plot_predict_interaction(forest, Boston, "rm", "lstat")
p + labs("my custom x lab", "my custom y lab")
It would be nice to be able to change x and y labels like you can with the title
The text was updated successfully, but these errors were encountered: