Sklearn Joblib与n_jobs

2024-09-28 20:44:36 发布

您现在位置:Python中文网/ 问答频道 /正文

使用scikit learn parallel_后端时,可以通过n个作业指定要运行的作业数:

from sklearn.externals import joblib

with joblib.parallel_backend(n_jobs=2):
    searchcv_1.fit()

但是,由于我有多个核心,我想运行另一个独立的GridSearchCV:

^{pr2}$

当然,这仍然是按顺序执行每个GridSearchCV,而不是为每个GridSearhCV分配两个核心。在

我真正想要的是在不同的核心上执行两个GridSearchCV命令。我怎么能同时做到这一点呢?在

from sklearn.externals import joblib

with joblib.parallel_backend(n_jobs=2):
    searchcv_1.fit(X1, Y1)

# This is still sequential after searchcv_1.fit
with joblib.parallel_backend(n_jobs=2):
    searchcv_2.fit(X2, Y1)

Tags: fromimportbackend核心parallelwith作业jobs