如何使用 sklearn 并行拟合梯度提升模型?

mow*_*per 5 python machine-learning scikit-learn

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

小智 0

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

您可以在这里阅读更多信息: https://scikit-learn.org/stable/modules/ensemble.html https://scikit-learn.org/stable/modules/computing.html#parallelism