spark updates for rf

Author: topepo

R/rand_forest.R

@@ -159,6 +159,11 @@ translate.rand_forest <- function(x, engine, ...) {
       )
     else
       x$method$fit_args$type <- x$mode
+
+    # See "Details" in ?ml_random_forest_classifier
+    if (is.numeric(x$method$fit_args$feature_subset_strategy))
+      x$method$fit_args$feature_subset_strategy <-
+        paste(x$method$fit_args$feature_subset_strategy)
   }
 
   # add checks to error trap or change things for this method

R/rand_forest_constr.R

@@ -39,7 +39,7 @@ rand_forest_randomForest_fit <-
     libs = "randomForest",
     interface = "data.frame",
     protect = c("x", "y"),
-    fit_name = c(pkg = "randomForest", fun = "randomForest.default"),
+    fit_name = c(pkg = "randomForest", fun = "randomForest"),
     alternates =
       list()
   )
@@ -49,7 +49,7 @@ rand_forest_spark_fit <-
   list(
     libs = "sparklyr",
     interface = "spark",
-    protect = c("x", "features_col", "label_col", "type"),
+    protect = c("x", "formula", "type"),
     fit_name = c(pkg = "sparklyr", fun = "ml_random_forest"),
     alternates =
       list(