Change x label in sv_waterfall() and sv_force()

ModelOriented · Sep 8, 2024 · 8ec6c00 · 8ec6c00
1 parent 8c04dd6
commit 8ec6c00
Show file tree

Hide file tree

Showing 5 changed files with 376 additions and 389 deletions.
diff --git a/NEWS.md b/NEWS.md
@@ -1,5 +1,9 @@
 # shapviz 0.9.5
 
+### User-visible changes
+
+- `sv_waterfall()` and `sv_force()`: The x label has been changed from "SHAP value" to "Prediction".
+
 ### Documentation
 
 - Add vignette for Tidymodels.

diff --git a/R/sv_force.R b/R/sv_force.R
@@ -113,7 +113,7 @@ sv_force.shapviz <- function(object, row_id = 1L, max_display = 6L,
       axis.ticks.y = ggplot2::element_blank(),
       axis.text.y = ggplot2::element_blank()
     ) +
-    ggplot2::labs(y = ggplot2::element_blank(), x = "SHAP value")
+    ggplot2::labs(y = ggplot2::element_blank(), x = "Prediction")
 
   if (show_annotation) {
     p <- p +

diff --git a/R/sv_waterfall.R b/R/sv_waterfall.R
@@ -126,7 +126,7 @@ sv_waterfall.shapviz <- function(object, row_id = 1L, max_display = 10L,
       axis.line.x = ggplot2::element_line(),
       axis.ticks.y = ggplot2::element_blank()
     ) +
-    ggplot2::labs(y = ggplot2::element_blank(), x = "SHAP value")
+    ggplot2::labs(y = ggplot2::element_blank(), x = "Prediction")
 
   if (show_connection) {
     p <- p +