@@ -7,7 +7,8 @@ library(tidymodels)
77library(usethis )
88
99# also requires installation of:
10- packages <- c(" parsnip" , " discrim" , " plsmod" , " rules" , " baguette" , " poissonreg" , " modeltime" , " modeltime.gluonts" )
10+ packages <- c(" parsnip" , " discrim" , " plsmod" , " rules" , " baguette" , " poissonreg" ,
11+ " multilevelmod" , " modeltime" , " modeltime.gluonts" )
1112
1213# ------------------------------------------------------------------------------
1314
@@ -63,6 +64,9 @@ get_tunable_param <- function(mode, package, model, engine) {
6364
6465model_db <-
6566 purrr :: map_dfr(packages , print_methods ) %> %
67+ dplyr :: filter(engine != " liquidSVM" ) %> %
68+ dplyr :: filter(model != " surv_reg" ) %> %
69+ dplyr :: filter(engine != " spark" ) %> %
6670 dplyr :: filter(! is.na(engine )) %> %
6771 dplyr :: mutate(label = paste0(model , " (" , engine , " )" )) %> %
6872 dplyr :: arrange(model , engine , mode )
@@ -77,7 +81,6 @@ num_modes <-
7781
7882model_db <-
7983 dplyr :: left_join(model_db , num_modes , by = c(" package" , " model" , " engine" )) %> %
80- dplyr :: filter(engine != " spark" ) %> %
8184 dplyr :: mutate(parameters = purrr :: pmap(list (mode , package , model , engine ), get_tunable_param ))
8285
8386usethis :: use_data(model_db , overwrite = TRUE )
0 commit comments