don't try to prune the root

bips-hb · May 16, 2024 · 2ecf6a0 · 2ecf6a0
1 parent 6101365
commit 2ecf6a0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/R/adversarial_rf.R b/R/adversarial_rf.R
@@ -195,6 +195,10 @@ adversarial_rf <- function(
       leaves <- which(out[[1]] == 0L)
       to_prune <- leaves[!(leaves %in% which(tabulate(pred[, tree]) >= min_node_size))]
       while(length(to_prune) > 0) {
+        if (1 %in% to_prune) {
+          # Never prune the root
+          break
+        }
         for (tp in to_prune) {
           # Find parents
           parent <- which((out[[1]] + 1L) == tp)