Scikit-Learn:RandomForestClassifier太大

时间:2018-08-22 21:59:32

标签: scikit-learn pickle

我正在使用Scikit-Learn在数据集上训练RandomForestClassifier。如果使用cPickle模块(默认设置)进行酸洗,得到的模型将在我的计算机上占用完整的14GB。

如果我用最高的协议来腌制,我会把它降低到4GB,这仍然太大了。

我正试图用它来预测新样品。结果,我只想将信息保留在预测所需的RandomForestClassifier对象中。在Scikit-Learn中有没有办法做到这一点?

要清楚一点,我不想更改任何超参数(例如max_depth)。我只想扔掉预测所不需要的信息。

谢谢!

0 个答案:

没有答案