Skip to main content
 首页 » 编程设计

python之拟合 RandomForestClassifier 时内存使用量激增

2025年05月04日10kenshinobiy

我正在尝试用中等大小的 numpy float 组来填充森林

In [3]: data.shape 
Out[3]: (401125, 5) 
 
[...] 
forest = forest.fit(data[:,1:],data[:,0]) 

此问题大约在开始拟合 10 秒后出现。在卡住我的机器之前它占用了多达 3.6 GB(3.8 GB)的内存,我不得不终止进程。

我一直在研究,当您将 n_jobs 设置为大于 1 的值时,这个问题似乎很常见。但我使用的是 n_jobs 的默认值,即 1。

我还尝试将 n_estimators 从我的初始值 100 更改为 5,只是为了看看是否发生了一些不同的事情,但一切都保持不变。

任何人都可以阐明这一点吗?

谢谢!

请您参考如下方法:

糟糕。我实际上是在尝试进行回归,而不是分类。 结束问题。