Fix bug [skip ci]

liao961120 · Jul 24, 2024 · 421c619 · 421c619
1 parent 91a7b48
commit 421c619
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/R/plot_forest.R b/R/plot_forest.R
@@ -82,11 +82,17 @@ plot_forest = function(dat, labels, shade_inv=c(.25,.75),
         } else if (center[1] == "mode") {
             center_pos = which(y == max(y))[1]
         } else {
-            x_sq_err = (x - mean(x))^2
-            center_pos = which(x_sq_err == min(x_sq_err))[1]
+            center_pos = -1
         }
 
-        mx = x[center_pos]; my = y[center_pos]
+        if (center_pos == -1) {
+            mx = mean(dat[[i]])
+            x_err = abs(x - mx)
+            center_pos = which(x_err == min(x_err))
+            my = y[center_pos]
+        } else {
+            mx = x[center_pos]; my = y[center_pos]
+        }
 
         # Shade center portion
         if (is.numeric(shade_inv)[1]) {
@@ -114,6 +120,7 @@ plot_forest = function(dat, labels, shade_inv=c(.25,.75),
     axis(2, at = y_centers, labels = labels, las=1)
 }
 
+
 is_odd = function(x) ifelse(x %% 2 == 1, T, F)