XGBoost如何进行并行计算?

时间:2015-12-08 08:21:00

标签: xgboost

XGBoost使用加法训练的方法,其中模拟先前模型的残差。

这是顺序的,然后如何并行计算呢?

1 个答案:

答案 0 :(得分:26)

Xgboost没有像你提到的那样并行运行多棵树,你需要在每棵树之后进行预测来更新渐变。

而是使用openMP在单个树中进行并行化,以独立创建分支。

要观察这一点,请构建一个巨型数据集并使用n_rounds = 1运行。您将看到所有核心在一棵树上触发。这就是它如此快速精心设计的原因。