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

时间:2015-04-21 08:02:52

标签: r extract rpart coefficients

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

1 个答案:

答案 0 :(得分:0)

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

threshold =  fit$split[1,foo]