提问人:Jim Chen 提问时间:10/30/2023 最后编辑:desertnautJim Chen 更新时间:11/2/2023 访问量:40
如何限制sklearn GridSearchCV CPU使用率?
How to limit sklearn GridSearchCV cpu usage?
问:
我使用如下: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 仍然使用我的所有内核:
我也尝试过,因为 sklearn 文档说
“对于低于 -1 的n_jobs,使用 (n_cpus + 1 + n_jobs)”n_jobs=-19
但仍然无法正常工作,我所有的 cpu 都被使用了。
我应该如何修改代码以降低 CPU 使用率?
答:
0赞
Jim Chen
11/1/2023
#1
事实证明,我必须设置为 lgbm。Regressor 参数,因此总 CPU 使用率为:n_jobs
total_cpu
= GridSeaechCV.fit n_jobs
xlgbm.fit n_jobs
评论