Pass extra arguments mice.impute.rf() to ranger::ranger()

edbonneville · Jun 28, 2023 · 222df22 · 222df22
1 parent 3f43b9a
commit 222df22
Show file tree

Hide file tree

Showing 2 changed files with 10 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,12 @@ 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)
+  extra_ranger_args <- list(...)
+  extra_ranger_args[["type"]] <- NULL # avoids unused argument warnings
+  fit <- do.call(
+    what = ranger::ranger,
+    args = c(list("x" = xobs, "y" = yobs, "num.trees" = ntree), extra_ranger_args)
+  )
 
   nodes <- predict(
     object = fit, data = rbind(xobs, xmis),

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