r - 识别派对 ctree 节点内的所有不同变量

Question

我正在 party R 包中使用 ctree 函数。我想识别树中使用的所有预测变量，以减少用于进一步分析的 data.frame 维度。例如:

library(ctree)
data(ozone)
myModel<-ctree(Ozone~., data=na.omit(airquality))
plot(myModel)

我想要一个接收 myModel 并返回温度、风和臭氧的函数

Answer 1

为了完整起见:NicE 的答案与 party 包中的 ctree() 实现有关。如果有人想要基于 partykit 包中新的(推荐的)实现做同样的事情，那么就需要一个不同的函数，因为内部表示完全改变了。

getUsefulPredictors <- function(x) {
  varid <- nodeapply(x, ids = nodeids(x),
    FUN = function(n) split_node(n)$varid)
  varid <- unique(unlist(varid))
  names(data_party(x))[varid]
}

这首先从树的每个节点中的每个分割中获取变量 ID varid。然后获取模型框架的名称并返回与唯一变量 ID 相关的名称。在您的示例中:

library("partykit")
myModel <- ctree(Ozone ~ ., data = na.omit(airquality))
getUsefulPredictors(myModel)    
## [1] "Temp" "Wind"