From 89e63db91ad6de963da589c386083183d60db71c Mon Sep 17 00:00:00 2001 From: Julia Silge Date: Fri, 20 Mar 2020 11:01:26 -0600 Subject: [PATCH] Skeleton idea --- R/rand_forest.R | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/R/rand_forest.R b/R/rand_forest.R index c1ea47e3b..33657edff 100644 --- a/R/rand_forest.R +++ b/R/rand_forest.R @@ -220,6 +220,13 @@ translate.rand_forest <- function(x, engine = x$engine, ...) { arg_vals$probability <- TRUE } + + ## for dimensionality hardening + mtry <- call2("min", + rlang::eval_tidy(arg_vals$mtry), + rlang::expr(.preds())) + arg_vals$mtry <- mtry + x$method$fit$args <- arg_vals x