【问题标题】:How can I extract decisions from a tree object of the R package rpart?如何从 R 包 rpart 的树对象中提取决策?
【发布时间】:2015-06-28 06:22:43
【问题描述】:

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

【问题讨论】:

    标签: r extract rpart coefficients


    【解决方案1】:

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

    threshold =  fit$split[1,foo]
    

    【讨论】:

      猜你喜欢
      • 2018-10-21
      • 2016-07-24
      • 1970-01-01
      • 2013-02-09
      • 2021-05-30
      • 2018-08-26
      • 2016-07-23
      • 2015-01-08
      • 2020-10-05
      相关资源
      最近更新 更多