1

我使用该包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$ 需要小于某个阈值。如何提取此阈值?我想在数据集上进行引导运行并将阈值保存在数组中。

4

1 回答 1

0

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

threshold =  fit$split[1,foo]
于 2015-04-21T09:01:43.373 回答