mshapviz importance plot color bar

ModelOriented · Sep 10, 2024 · 39fb917 · 39fb917
1 parent bc61993
commit 39fb917
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/R/plot_utils.R b/R/plot_utils.R
@@ -24,11 +24,11 @@
 }
 
 # Slimmer colorbars with less space
-.slim_colorbar <- function(height = 1.2) {
+.slim_colorbar <- function(height = 1.2, width = 0.5) {
   ggplot2::theme(
     legend.box.spacing = grid::unit(0, "lines"),
     legend.key.spacing = grid::unit(0.3, "lines"),
-    legend.key.width = grid::unit(0.5, "lines"),
+    legend.key.width = grid::unit(width, "lines"),
     legend.key.height = grid::unit(height, "lines"),
     legend.ticks.length = ggplot2::rel(0.3)
   )

diff --git a/R/sv_importance.R b/R/sv_importance.R
@@ -198,7 +198,8 @@ sv_importance.mshapviz <- function(object, kind = c("bar", "beeswarm", "both", "
         ) +
         ggplot2::labs(fill = ggplot2::element_blank()) +
         do.call(ggplot2::scale_fill_viridis_d, viridis_args) +
-        ggplot2::guides(fill = ggplot2::guide_legend(reverse = TRUE))
+        ggplot2::guides(fill = ggplot2::guide_legend(reverse = TRUE)) +
+        .slim_colorbar(width = 1)
     } else {  # facets
       p <- ggplot2::ggplot(imp_df, ggplot2::aes(x = values, y = feature)) +
         ggplot2::geom_bar(fill = fill, width = bar_width, stat = "identity", ...) +

diff --git a/man/figures/VIGNETTE-tidy-class-xgb-imp.png b/man/figures/VIGNETTE-tidy-class-xgb-imp.png