我正在使用Scikit-Learn
在数据集上训练RandomForestClassifier
。如果使用cPickle
模块(默认设置)进行酸洗,得到的模型将在我的计算机上占用完整的14GB。
如果我用最高的协议来腌制,我会把它降低到4GB,这仍然太大了。
我正试图用它来预测新样品。结果,我只想将信息保留在预测所需的RandomForestClassifier
对象中。在Scikit-Learn
中有没有办法做到这一点?
要清楚一点,我不想更改任何超参数(例如max_depth
)。我只想扔掉预测所不需要的信息。
谢谢!