#120

easystats · Jun 10, 2021 · 675133d · 675133d
1 parent af5668d
commit 675133d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/R/visualisation_recipe.estimate_predicted.R b/R/visualisation_recipe.estimate_predicted.R
@@ -168,7 +168,11 @@ visualisation_recipe.estimate_predicted <- function(x,
     if (insight::model_info(info$model)$is_binomial && show_data %in% c("point", "points")) {
       shape <- "|"
       stroke <- 1
-      # TODO: adjust in the case of non 0-1 rawdata[[y]]
+
+      # Change scale to 1-2 in case outcome is factor (see #120)
+      if(!all(unique(rawdata[[y]]) %in% c(0, 1))) {
+        data[c("Predicted", "CI_low", "CI_high")] <- data[c("Predicted", "CI_low", "CI_high")] + 1
+      }
     }
 
     for (i in show_data) {