10x speedup because earth does cv for no reason

tlverse · Jan 6, 2023 · afc4d66 · afc4d66
1 parent 85b87fa
commit afc4d66
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/R/screening.R b/R/screening.R
@@ -398,6 +398,9 @@ fit_earth_hal <- function(X,
 #' screen_MARS(X, Y, degree = 1)
 #'
 screen_MARS <- function(x, y, pmethod = "cv", degree = 2, nfold = 10, fast.k = NULL, nk = NULL, glm = list(family = gaussian()), weights = NULL) {
+  if(pmethod != "cv") {
+    nfold <- 0
+  }
   X <- x
   Y <- y
   n <- length(Y)