如何将渐变增强模型与sklearn并行化?

时间:2017-09-02 15:38:31

标签: python machine-learning scikit-learn

使用sklearn进行模型构建相对较新。我知道交叉验证可以通过n_jobs参数进行并行化,但是如果我不使用CV,我如何利用我的可用内核来加速模型拟合?

2 个答案:

答案 0 :(得分:1)

XGboost 或 LigtGMB 等替代方案是分布式的(即可以并行运行)。这些都是有据可查的流行提升算法。

答案 1 :(得分:0)

目前afaik的GradientBoosting方法没有像“ n_jobs”这样的参数,但是您可以尝试使用HistGradientBoostingRegressor,它将OpenMP用于并行化(将使用尽可能多的线程)。

您可以在此处了解更多信息: https://scikit-learn.org/stable/modules/ensemble.html https://scikit-learn.org/stable/modules/computing.html#parallelism