Question

我有很多树，打印时它们长7页。我不得不重新平衡数据，需要查看频率最高的分支，看看它们是否有意义 - 我需要确定不同群集的取消率。

鉴于数据太长，我需要的是拥有最大的分支，然后我可以验证这些而不是手动通过210个分支。我会有很多树，所以需要自动化这些以查看重要的结果。

使用示例代码：

library(CHAID)
updatecars<-mtcars
updatecars$cyl<-as.factor(updatecars$cyl)
updatecars$vs<-as.factor(updatecars$vs)
updatecars$am<-as.factor(updatecars$am)
updatecars$gear<-as.factor(updatecars$gear)
plot(carsChaid)

carsChaid<-chaid(am~  cyl+vs+gear, data=updatecars)
carsChaid

当您打印此数据时，您会看到第一组的n = 15。我需要一个表格，我可以对这个值进行排序。

我需要的是一个决策树表，其中包含变量值和树中每个组内的数字。这与此答案Walk a tree不完全相同因为它没有给出数字，但我认为它是方向。

有人可以帮忙吗，
谢谢，
詹姆斯

Answer 1

当然有一种更好的方法可以做到这一点，但这样做。显然愿意进行更正和改进。

我遇到的特别麻烦是创建所有组合的列表。当expand.grid超过3个因素时，它将停止工作。所以我不得不在它上面构建一个循环来创建完整的列表。

All_canx_rates<-function(Var1,Var2,Var3,Var4,Var5,nametree){
  df1<-data.frame("CanxRate"=0,"Num_Canx"=0,"Num_Cust"=0)
  pars<-as.list(match.call()[-1])
  a<-eval(pars$nametree)[,as.character(pars$Var1)]
  b<-eval(pars$nametree)[,as.character(pars$Var2)]
  c<-eval(pars$nametree)[,as.character(pars$Var3)]
  d<-eval(pars$nametree)[,as.character(pars$Var4)]
  e<-eval(pars$nametree)[,as.character(pars$Var5)]

  allcombos<-expand.grid(levels(a),levels(b),levels(c))
  clean<- allcombos
  allcombos$Var4<-d[1]

  for (i in 2:length(levels(d))) {
    clean$Var4<-levels(d)[i]  
    allcombos<-rbind(allcombos,clean)
  }

  #define a forloop
  for (i in 1:nrow(allcombos)) {
    #define values
    f1<-allcombos[i,1]
    f2<-allcombos[i,2]
    f3<-allcombos[i,3]
    f4<-allcombos[i,4]

  y5<-nrow(nametree[(a %in% f1 & b %in% f2 & c %in% f3 & d %in% f4 & 
                       e =='1'),])
  y4<-nrow(nametree[(a %in% f1 & b %in% f2 & c %in% f3 & d %in% f4),])
  df2<-data.frame("CanxRate"=y5/y4,"Num_Canx"=y5,"Num_Cust"=y4)
  df1<-rbind(df1, df2)

  }
  #endforloop
  #make the dataframe available for global viewing
  df1<-df1[-1,]
  output<<-cbind(allcombos,df1)
  }

Answer 2

您可以使用data.tree对派对对象进行进一步操作，例如排序，遍历树，自定义绘图等。来自github的最新版本v0.3.7具有来自派对类对象的转换：

devtools::install_github("gluc/data.tree@v0.3.7")
library(data.tree)
tree <- as.Node(carsChaid)

tree$fieldsAll

最后一个命令显示派对类的转换字段的名称：

[1] "data"        "fitted"      "nodeinfo"    "partyinfo"   "split"       "splitlevels" "splitname"   "terms"       "splitLevel"

您可以按功能排序，例如每个节点上的数据行：

tree$Sort(attribute = function(node) nrow(node$data), decreasing = TRUE)

print(tree, 
      "splitname",
      count = function(node) nrow(node$data), 
      "splitLevel")

例如，这样打印：

  levelName splitname count splitLevel
1     1          gear    32           
2      ¦--3              17       4, 5
3      °--2              15          3

走一个CHAID树R - 需要按实例数排序

2 个答案: