识别政党ctree模型中所有不同的变量。

Question

我在R方包中使用ctree函数。为了减少用于进一步分析的data.frame维数，我希望对树中使用的所有预测器进行验证。例如：

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

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

Answer 1

您可以尝试使用以下方法：

getUsefulPredictors<-function(x){
  flatTree<-unlist(x@tree)
  pred<-unique(flatTree[grepl("*variableName",names(flatTree))])
  return(pred)
}

它将树夷为平地，并寻找以variableName命名的元素。

在它返回的模型上运行：

getUsefulPredictors(myModel)
#[1] "Temp" "Wind"

Answer 2

为了完整起见: 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"