【发布时间】:2015-10-15 03:48:10
【问题描述】:
我有 sklearn 随机森林回归器。它非常重,1.6 GB,并且在预测值时工作很长时间。
我想修剪它以使其更轻。据我所知,对决策树和森林没有实施修剪。我自己无法实现,因为树代码是用 C 编写的,我不知道。
有人知道解决办法吗?
【问题讨论】:
-
我认为你应该限制树的大小(最大叶节点,最大深度,最小样本分割......)
-
invoketheshell,谢谢你的链接。主要思想是使用森林的并行状态来使用所有 CPU 内核。就我而言,它已经完成了。
标签: python machine-learning scikit-learn random-forest pruning