Restoring enmtools.vip plots for shapley

danlwarren · Aug 6, 2024 · d8ad17c · d8ad17c
1 parent 683272b
commit d8ad17c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/enmtools.vip.R b/R/enmtools.vip.R
@@ -186,7 +186,7 @@ enmtools.vip <- function(model, metric = "roc_auc", nsim = 10, method = "permute
 
     output[["shap.plot"]] <- ggplot(output[["shap"]],
                                     aes_string(x = "Importance",
-                                               fill = after_stat("..x.."))) +
+                                               fill = fct_reorder("Variable", "Importance", .desc = TRUE))) +
       geom_histogram(bins = 20) +
       theme_bw() +
       geom_hline(yintercept = 0, color = "grey") +
@@ -239,7 +239,7 @@ enmtools.vip <- function(model, metric = "roc_auc", nsim = 10, method = "permute
       geom_histogram(bins = 20) +
       theme_bw() +
       geom_hline(yintercept = 0, color = "grey") +
-      viridis::scale_fill_viridis(name = "Variable", option = "D", discrete = TRUE, direction = -1) +
+      viridis::scale_fill_viridis(name = "Variable", option = "D", discrete = TRUE) +
       facet_grid(rows = vars(fct_reorder(.data$Variable, .data$Importance, .desc = TRUE)), switch = "y") +
       ylab("Variable") +
       ggtitle("Variable importance, FIRM method") +