如何从 R 包 rpart 的树对象中提取决策



我使用包rpart来建模分类/回归树。我有变量 x,y,s,其中 x 在 {-1,1} 中,y 在 [0,1] 中连续,s$ 是一个具有 3 个水平的因子。

我使用

fit <- rpart(x~y+s, data=data, method="class")

最终模型非常有意义,我可以使用fancyRpartPlot(fit)绘制它。我也可以修剪树,只剩下一个决定:

fit = prune(fit,cp=0.1) 

它说$y$需要小于某个阈值。如何提取此阈值?我想在数据集上进行引导运行并将阈值保持在数组中。

rpart对象中split字段包含决策边界。解决方案是

threshold =  fit$split[1,foo]

最新更新