Skip to content

Commit e7e6f6e

Browse files
authoredMar 16, 2022
update add-in info (#673)
1 parent f89cdea commit e7e6f6e

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed
 

‎data/model_db.rda

751 Bytes
Binary file not shown.

‎inst/add-in/parsnip_model_db.R

+5-2
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,8 @@ library(tidymodels)
77
library(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

6465
model_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

7882
model_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

8386
usethis::use_data(model_db, overwrite = TRUE)

0 commit comments

Comments
 (0)