dask - 为一个函数提供多个核心而不是 1 个

Question

假设我有 8 个内核，我想运行 2 个函数，每个函数都使用 4 个内核（例如 ML 模型），最好的设置方法是什么。我在文档中找不到这样的例子。

例如，让我们说 sklearn 随机森林来保持简单。

def our_func(X, y):
    return RandomForestClassifier(n_jobs=4).fit(X, y)

a = client.submit(our_func, X1, y1)  
b = client.submit(our_func, X2, y2)

我如何确保函数将使用 4 个不同的进程？

标签： dask