Merge pull request #563 from edbonneville/passing_args_ranger

Pass extra arguments mice.impute.rf() to ranger::ranger()
amices · Jul 19, 2023 · 87073f8 · 87073f8
2 parents 3f43b9a + f1c697b
commit 87073f8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/R/mice.impute.rf.R b/R/mice.impute.rf.R
@@ -10,8 +10,8 @@
 #' alternative currently implemented is the \code{randomForest} package, which
 #' used to be the default in mice 3.13.10 and earlier.
 #' @param \dots Other named arguments passed down to
-#' \code{mice:::install.on.demand()}, \code{randomForest::randomForest()} and
-#' \code{randomForest:::randomForest.default()}.
+#' \code{mice:::install.on.demand()}, \code{randomForest::randomForest()},
+#' \code{randomForest:::randomForest.default()}, and \code{ranger::ranger()}.
 #' @return Vector with imputed data, same type as \code{y}, and of length
 #' \code{sum(wy)}
 #' @details
@@ -106,7 +106,7 @@ mice.impute.rf <- function(y, ry, x, wy = NULL, ntree = 10,
   install.on.demand("ranger", ...)
 
   # Fit all trees at once
-  fit <- ranger::ranger(x = xobs, y = yobs, num.trees = ntree)
+  fit <- suppressWarnings(ranger::ranger(x = xobs, y = yobs, num.trees = ntree, ...))
 
   nodes <- predict(
     object = fit, data = rbind(xobs, xmis),

diff --git a/man/mice.impute.rf.Rd b/man/mice.impute.rf.Rd