如何在tidymodels中为boost_tree()插入正则化参数？在普通的lightgbm包中有调谐参数lambda_l1.我也想用tidymodels个.
我试着这样写，但我不确定我是否做得对:
lgbm_model <-
boost_tree(
mode = "regression",
# mtry = 1,
trees = tune(),
min_n = tune(),
tree_depth = tune(),
learn_rate = tune(),
loss_reduction = tune()
) %>%
set_engine("lightgbm", lambda = 1)
由于lambda_l1不是boost_tree()的主要参数，所以你确实会把这个参数提供给set_engine().按照lightgbm::lgb.train()或param参数中的名称提供它，使用lambda_l1.bonsai包实现了对"lightgbm"引擎的支持，它将负责将参数传递到正确的位置.
你的代码看起来像这样:
library(tidymodels)
library(bonsai)
lgb <-
boost_tree(mode = "regression",) %>%
set_engine("lightgbm", lambda_l1 = .9)
lgb_fit <- fit(lgb, mpg ~ ., mtcars)
lgb_fit
#> parsnip model object
#>
#> LightGBM Model (1 tree)
#> Objective: regression
#> Fitted to dataset with 10 columns
为了确认lambda_l1是按您的预期提供的，您可以提取底层LightGBM fit并戳其中:
lgb_fit_engine <- extract_fit_engine(lgb_fit)
lgb_fit_engine$params$lambda_l1
#> [1] 0.9
创建于2024—03—28，reprex v2.1.0
:)