【问题标题】:Extract probabilities from decision trees从决策树中提取概率
【发布时间】:2015-01-13 18:08:45
【问题描述】:

我需要提取决策树中每个叶子的路径和概率。这是一个可以使用的快速示例:

data(iris)

model<-rpart(Species~., data=iris)

summary(model)

我希望能够以一种我可以放入表格而不是仅仅能够在控制台上阅读。

非常感谢!

【问题讨论】:

  • unclass(model) 有很多可能有用的东西。看起来如果你挑出你想要的东西,摆出来可能还不错

标签: r decision-tree


【解决方案1】:

您需要访问“模型”的属性,它是 rpart.object 的实例。 文件: https://stat.ethz.ch/R-manual/R-devel/library/rpart/html/rpart.object.html

例如槽 "where" 有以下信息: 一个与根节点中的观察个数等长的整数向量,包含每个观察落入的叶子节点对应的帧的行号。

你会在槽“frame”中找到的概率。

【讨论】:

  • 这可能是正确的方法。我将不得不盯着它看一会儿,因为它不是很直观。他们在 model$frame$yval2 中列出的概率与 summary(model) 中给出的概率不同
  • 我在 model$frame 的以下列中看到相同的概率:yval2.V5, yval2.V6, yval2.V7
  • 不敢相信我花了多长时间才找到这些,谢谢!
猜你喜欢
  • 2014-06-09
  • 2020-12-03
  • 2012-08-04
  • 2019-08-07
  • 2019-05-24
  • 2021-12-16
  • 1970-01-01
  • 2013-12-12
相关资源
最近更新 更多