tidymodels arguments with submodels

arguments_with_submodels.md

library(tidyverse)
library(parsnip)
x <- lapply(parsnip:::extensions(), require, character.only = TRUE)
#> Loading required package: baguette
#> Loading required package: censored
#> Loading required package: survival
#> Loading required package: discrim
#> Loading required package: multilevelmod
#> Loading required package: plsmod
#> Loading required package: poissonreg
#> Loading required package: rules
#> Loading required package: bonsai
#> Loading required package: agua

parsnip::get_model_env() %>%
  as.list() %>%
  enframe() %>%
  filter(grepl("args", name)) %>%
  mutate(name = gsub("_args", "", name)) %>%
  unnest(value) %>%
  filter(has_submodel) %>%
  select(model = name, engine, argument = parsnip)
#> # A tibble: 15 × 3
#>    model                engine   argument 
#>    <chr>                <chr>    <chr>    
#>  1 rule_fit             xrf      penalty  
#>  2 logistic_reg         glmnet   penalty  
#>  3 poisson_reg          glmnet   penalty  
#>  4 pls                  mixOmics num_comp 
#>  5 discrim_flexible     earth    num_terms
#>  6 multinom_reg         glmnet   penalty  
#>  7 mars                 earth    num_terms
#>  8 C5_rules             C5.0     trees    
#>  9 cubist_rules         Cubist   neighbors
#> 10 proportional_hazards glmnet   penalty  
#> 11 boost_tree           xgboost  trees    
#> 12 boost_tree           C5.0     trees    
#> 13 boost_tree           lightgbm trees    
#> 14 linear_reg           glmnet   penalty  
#> 15 nearest_neighbor     kknn     neighbors

^{Created on 2023-10-11 with reprex v2.0.2}