使用XGBoos中的基树

2024-06-16 11:43:29 发布

您现在位置:Python中文网/ 问答频道 /正文

有没有办法只使用XGBoost算法中使用的基本决策树?你知道吗

我知道Sklearn的GBT只是使用Sklearn决策树作为它们的基础,但是XGBoost构建树的方式不同(例如,叶权重的正则化)。你知道吗

我查看了XGBoost的代码,但无法理解它们是如何构建基树的。你知道吗

谢谢!你知道吗


Tags: 代码算法决策树方式sklearn基础权重正则
1条回答
网友
1楼 · 发布于 2024-06-16 11:43:29

有两种解决方案:

  1. 只构建一棵树(意味着只使用一次迭代),这意味着在训练时使用num_boost_round=1
  2. 预测时使用ntree_limit=1

任何一个都能解决你的问题。你要么只训练一棵树,要么只使用一棵树进行预测。你知道吗

相关问题 更多 >