gpt4 book ai didi

scikit-learn - sklearn随机森林未并行化

转载 作者:行者123 更新时间:2023-12-02 05:17:18 26 4
gpt4 key购买 nike

我在Ubuntu 12.04上使用sklearn 0.16并运行:

from sklearn.ensemble import RandomForestClassifier
import numpy as np
X=np.random.rand(5000,500)
y=(np.random.rand(5000).round())
RandomForestClassifier(n_jobs=10,n_estimators=1000).fit(X,y)

但是,它没有用尽我的核心,并且花费与n_jobs = 1相同的时间。关于如何调试这里发生的事情有什么想法吗?

此屏幕快照显示了其他正在运行的事情,但是htop一直在显示可用的CPU:

最佳答案

尝试这个 :

import affinity
import multiprocessing
affinity.set_process_affinity_mask(0, 2**multiprocessing.cpu_count()-1)

关于scikit-learn - sklearn随机森林未并行化,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/29322005/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com