整洁模型中XGBoost的L2正则化

Question

我正在使用tidymodels包来使用XGBoost模型。

我的型号是

boost_tree(mtry = tune(), trees = 1000, learn_rate = tune()) 
%>% set_engine('xgboost', objective = 'binary:hinge' )
%>% set_mode('classification')

我想知道把L2正则化超参数lambda放在哪里。

Rxgboost文档列出lambda术语，

但是在tidymodels的boost_tree文档中没有lambda项。

Answer 1

您可以将任何特定于引擎的参数放入set_engine()中，例如%>% set_engine('xgboost', objective = 'binary:hinge', lambda = 2)。