r语言 - lightgbm engine in tidymodels

Question

当我想将lightgbm引擎用于parsnip包中的boost_tree函数时，我会收到这样的错误：

lgbm_model <- 
boost_tree( mode = "classification",
mtry = tune(),
trees = tune(),
min_n = tune(),
tree_depth = tune(),
learn_rate = tune(),
loss_reduction = tune(),
engine = "lightgbm"
)

错误：

check_spec_mode_engine_val()中的错误：！boost_tree()不支持引擎"lightgbm"。参见show_engines('boost_tree')。运行rlang::last_error()查看错误发生的位置。

这个问题可能与什么有关？

Answer 1

show_engines('boost_tree')显示型号boost_tree():的可用引擎

发动机模式

1 xgboost分类2 xgboost回归
3 C5.0分类4火花分类5火花回归

正如我们所看到的，这里没有lightgbm引擎，因此它不能用作引擎输入。

解决方案是将新引擎添加到parsnip中。你可以在这里阅读关于如何做到这一点