From 70718e02734f8c89081537ba93bba59cd0819e05 Mon Sep 17 00:00:00 2001 From: Mike Tokic Date: Wed, 10 Apr 2024 11:30:14 -0700 Subject: [PATCH] add correct package to fit function --- R/multistep_glmnet.R | 2 +- R/multistep_mars.R | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/R/multistep_glmnet.R b/R/multistep_glmnet.R index 7ea01dad..0ab6f155 100644 --- a/R/multistep_glmnet.R +++ b/R/multistep_glmnet.R @@ -335,7 +335,7 @@ glmnet_multistep_fit_impl <- function(x, y, # fit model fit_glmnet <- linreg_reg_spec %>% - fit(Target ~ ., data = combined_df) + generics::fit(Target ~ ., data = combined_df) # create prediction glmnet_fitted <- predict(fit_glmnet, combined_df) diff --git a/R/multistep_mars.R b/R/multistep_mars.R index 2336b8e5..b4f04f06 100644 --- a/R/multistep_mars.R +++ b/R/multistep_mars.R @@ -358,7 +358,7 @@ mars_multistep_fit_impl <- function(x, y, # fit model fit_mars <- mars_reg_spec %>% - fit(Target ~ ., data = combined_df) + generics::fit(Target ~ ., data = combined_df) # create prediction mars_fitted <- predict(fit_mars, combined_df)