如何限制sklearn GridSearchCV cpu使用？

Question

我使用

GridSearchCV

如下：

gsearch_lgb = GridSearchCV(
    model(**self.model_params),
    param_grid=self.model_params,
    n_jobs=2,
    verbose=99,
    scoring=self.cv_scoring,
    cv=4,
)

但是 joblib 仍然使用我的所有核心：

我也尝试过

n_jobs=-19

，因为sklearn文档说 “对于低于 -1 的 n_jobs，使用 (n_cpus + 1 + n_jobs)”

但是还是不行，我的cpu都用完了。

我应该如何修改代码以减少 CPU 使用率？

Answer 1

原来我还必须将

n_jobs

设置为

lgbm.fit

，所以总 CPU 使用率将是：

GridSeaechCV.fit n_jobs * lgbm.fit n_jobs