GridSearchCV和Google Colab:n_jobs=-1无法正常工作。

3
问题如下:当运行以下代码时
grid = GridSearchCV(estimator=model, param_grid=param_grid, n_jobs=-1)
grid_result = grid.fit(X, y)

在谷歌 Colab 上,它返回错误。
PicklingError: Could not pickle the task to send it to the workers.

然而,如果我设置 n_jobs = 1,它可以完美运行。

我该如何在Google Colab上使用多个处理器?


请提供您的完整代码,好吗?您是否使用了其他数字,例如n_jobs=2 - Raha Moosavi
1个回答

3

你的问题可能存在以下几个可能性:

  • 一个问题可能是你的param_grid=param_grid,有时候改变网格搜索参数就可以解决问题。

  • 另一个潜在的问题可能与您的模型和使用的参数有关。参见这个问题作为例子

  • 您可以尝试任何其他数字,例如使用n_jobs=4而不是n_jobs=-1。此外,您还可以使用:

n_cpus = multiprocessing.cpu_count()

获取 CPU 数量,然后...

n_jobs=n_cpus 

或者

n_jobs=n_cpus - 1 

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接