@@ -7,7 +7,8 @@ library(tidymodels)
7
7
library(usethis )
8
8
9
9
# 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" )
11
12
12
13
# ------------------------------------------------------------------------------
13
14
@@ -63,6 +64,9 @@ get_tunable_param <- function(mode, package, model, engine) {
63
64
64
65
model_db <-
65
66
purrr :: map_dfr(packages , print_methods ) %> %
67
+ dplyr :: filter(engine != " liquidSVM" ) %> %
68
+ dplyr :: filter(model != " surv_reg" ) %> %
69
+ dplyr :: filter(engine != " spark" ) %> %
66
70
dplyr :: filter(! is.na(engine )) %> %
67
71
dplyr :: mutate(label = paste0(model , " (" , engine , " )" )) %> %
68
72
dplyr :: arrange(model , engine , mode )
@@ -77,7 +81,6 @@ num_modes <-
77
81
78
82
model_db <-
79
83
dplyr :: left_join(model_db , num_modes , by = c(" package" , " model" , " engine" )) %> %
80
- dplyr :: filter(engine != " spark" ) %> %
81
84
dplyr :: mutate(parameters = purrr :: pmap(list (mode , package , model , engine ), get_tunable_param ))
82
85
83
86
usethis :: use_data(model_db , overwrite = TRUE )
0 commit comments