Merge pull request #710 from imbs-hl/issue709

Add check for inbag element size
imbs-hl · Nov 22, 2023 · 24a24bf · 24a24bf
2 parents dcaee8d + b9792f2
commit 24a24bf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/R/ranger.R b/R/ranger.R
@@ -659,6 +659,9 @@ ranger <- function(formula = NULL, data = NULL, num.trees = 500, mtry = NULL,
     if (length(inbag) != num.trees) {
       stop("Error: Size of inbag list not equal to number of trees.")
     }
+    if (any(sapply(inbag, length) != nrow(x))) {
+      stop("Error: Size of at least one element in inbag not equal to number of samples.")
+    }
   } else {
     stop("Error: Invalid inbag, expects list of vectors of size num.trees.")
   }