如何从 `party` 包中的 `ctree()` 构建的回归树中删除某些节点答案

【问题标题】：How to delete certain nodes from a regression tree built by `ctree()` from `party` package如何从 `party` 包中的 `ctree()` 构建的回归树中删除某些节点
【发布时间】：2014-06-25 11:17:50
【问题描述】：

我使用包party 中的ctree() 构建了一个回归树。我的模型的结果有许多节点，其中包含相等概率的因变量（例如：A 类 = 0.33，B 类 = 0.33，C 类 = 0.33）。我想从模型中取出这些节点。包tree 有snip.tree() 命令，我们可以在其中指定要从模型中删除的节点号。该命令不识别使用ctree() 构建的回归树。请让我知道是否有办法从使用 ctree() 构建的回归树中删除某些节点

我用过模型：

rv.mod1 <- ctree(ldclas ~ L2 + L3 + L4 + L5 + L6 + ele + ndvi + nd_var + nd_ps, data = rv, controls = ctree_control(minsplit = 0, minbucket = 0))
pr.rv.mod1 <- snip.tree(rv.mod1, nodes = nn2.rv.mod1$nodes)

nn2.rv.mod1$nodes 是一个向量，其中包含要从 rv.mod1 模型中删除的节点。但是我收到一个错误：

Error in snip.tree(rv.mod1, nodes = nn2.rv.mod1$nodes) : 
  not legitimate tree

【问题讨论】：

你能告诉我们你做了什么吗？另外：请阅读how to ask a good question上的帮助页面

标签： r decision-tree party

【解决方案1】：

我不认为有直接的方法可以做到这一点，但我会在ctree 中使用weights 参数提出一个“hack”。

让我们从一个可重现的例子开始

library(party)
irisct <- ctree(Species ~ .,data = iris)
plot(irisct)

现在，假设您想摆脱节点号 5。您可以执行以下操作

NewWeigths <- rep(1, dim(iris)[1]) # Setting a weights vector which will be passed into the `weights` attribute in `ctree`
Node <- 5 # Selecting node #5
n <- nodes(irisct, Node)[[1]] # Retrieving the weights of that node
NewWeigths[which(as.logical(n$weights))] <- 0 # Setting these weigths to zero, so `ctree` will disregard them
irisct2 <- ctree(Species ~ .,data = iris, weights = NewWeigths) # creating the new tree with new weights
plot(irisct2)

请注意节点 2、6 和 7（现在它们被命名为 2、4 和 5，因为我们拆分较少）如何保持完全相同的分布和拆分条件。

我没有对所有节点进行测试，但它似乎工作得相当好

【讨论】：