r - party パッケージの ctree 関数 - 分割変数のリストを簡単な方法で取得する方法

Question

を使用しているときに、ツリーの分割変数のリストを取得しようとしていますctree。

psplit を使用して、特定のノードの特定の分割変数を取得する方法を見つけました。しかし、ツリー全体の分割変数のリスト全体を取得したいと思います。

取得する簡単な方法はありますか？

Answer 1

var.ctree<-function(fit){

  require(gdata)

  a<-capture.output(print(fit))  
  a<-a[-c(1:7)]   
  a<-trim(a)   

  v<-character()   
  for(h in 1:length(a)){
    b<-a[h]    
    b<-gsub(") ","q",b)    
    b<-gsub(")","q",b)    
    b<-gsub(" < ","q",b)    
    b<-gsub(" > ","q",b)    
    b<-gsub(" <= ","q",b)    
    b<-gsub(" <= ","q",b)    
    b<-gsub("  weights = ","q",b)    
    v[h]<-unlist(strsplit(b,"q"))[2]    
  }    
  v<-factor(v,levels=names(fit@data@get("input")))    
  v<-v[is.na(v)==FALSE]    
  tabla<-table(v)>0    
  sol<-as.numeric(table(v)>0)    
  names(sol)<-names(tabla)    
  sol    
}