python - GridSearchCV 和 Google colab : n_jobs=-1 does not work

Question

问题如下:当运行这些代码行时

grid = GridSearchCV(estimator=model, param_grid=param_grid, n_jobs=-1)
grid_result = grid.fit(X, y)

在 Google Colab 上，它返回错误

PicklingError: Could not pickle the task to send it to the workers.

但是，如果我设置 n_jobs = 1，它会完美地工作。

如何在 Google Colab 上使用多个处理器？

Answer 1

您的问题可能存在以下问题:

• 一个问题可能是您的 param_grid=param_grid，有时更改网格搜索参数可以解决问题。

• 另一个潜在问题可能与您的模型和其中使用的参数有关。将此问题视为 example .

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

n_cpus = multiprocessing.cpu_count()

获取CPU数量然后

n_jobs=n_cpus 

或

n_jobs=n_cpus - 1