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

Question

我正在尝试用中等大小的 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，只是为了看看是否发生了一些不同的事情，但一切都保持不变。

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

谢谢!

Answer 1

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