R:减小rpart对象的大小

时间:2015-12-17 10:30:46

标签: r machine-learning

我正在尝试在列表中聚合使用rpart::rpart构建的多个CART模型。

我刚刚意识到每个模型在$terms$where中存储了很多(元?)数据(在我的情况下每个模型超过10MB的数据),这使我的最终列表无法管理

在我看来,print(my_rpart_object)给出的摘要应该足以描述对象并运行预测,所以我想知道是否有方法来修剪/压缩rpart树?

2 个答案:

答案 0 :(得分:0)

找到它:每个paste0('000', TestID) 对象都带有一个环境。要删除它:

rpart

21个部分对象的列表从1.2GB到8MB。

答案 1 :(得分:0)

我也在为此苦苦挣扎。我发现将rpart树的“ where”元素设置为NULL可以大大减少树的内存占用量:

rpart_model <- rpart(...)
rpart_model$where <- NULL