Bug fix: Adjust row_lbl of 2x2 confusion matrix to read "Prediction" …

…when using "test" data, but keep default of "Decision" otherwise.
hneth · Sep 4, 2024 · 73d4128 · 73d4128
1 parent 7233196
commit 73d4128
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/R/plotFFTrees_function.R b/R/plotFFTrees_function.R
@@ -1640,7 +1640,7 @@ plot.FFTrees <- function(x = NULL,
             # print(lloc)  # 4debugging
 
 
-            # Classification table: ----
+            # Classification table: 2x2 matrix / confusion matrix: ----
 
             if (show.confusion) {
 
@@ -1695,16 +1695,19 @@ plot.FFTrees <- function(x = NULL,
                 decision.labels[1], adj = 1
               )
 
+              # Adjust row category label:
+              if (data == "test") {
+                row_lbl <- "Prediction"
+              } else { # default:
+                row_lbl <- "Decision"
+              }
+
               text(
                 x = final_classtable_x[1] - .065,
                 y = mean(final_classtable_y), cex = header_cex,
-                "Decision"
+                row_lbl
               )
 
-              # text(x = final_classtable_x[1] - .05,
-              #      y = mean(final_classtable_y), cex = header_cex,
-              #      "Decision", srt = 90, pos = 3)
-
 
               # Add final frequencies: ----