Sklearn 在多个函数中使用了 n_jobs
参数。我们可以传递想要使用的核心数量。例如,1 表示使用一个核心,-1 表示使用所有核心。我的系统有六个核心,我想只使用 [0,1,3,4] 这些核心。这是可能的吗?
回答:
我之前从未尝试过或听说过这种可能性,使用 scikit-learn
。
尽管如此,由于 scikit-learn
使用 joblib
,你可以考虑研究如何实现自定义的 ParallelBackendBase
,这是一个帮助定义所有 ParallelBackend
必须实现的方法的抽象基类 1, 2, 3;你可能能够在内部实现 CPU 亲和性,将进程固定到特定的 CPU 核心上。
然而,这完全是实验性的。