我使用包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运行并将阈值保持在数组中。
答案 0 :(得分:0)
split
对象中的字段rpart
包含决策边界。
解决方案是
threshold = fit$split[1,foo]