2017-03-22 9 views
1

にC5.0アルゴリズムのすべての予測行のルールを取得する私はデータにこのモデルを当てはめた後、道どのようにR

model <- c5.0(Target ~.,data=df,rules=TRUE,trials=10) 

を以下にC5.0 WTHモデルを適合しています、私はルールを持つようにしたいです予測された行ごとに

私は次の関数に

pathpred <- function(object, ...) { 
    ## coerce to "party" object if necessary 
    if(!inherits(object, "party")) object <- as.party(object) 

    ## get standard predictions (response/prob) and collect in data frame 
    rval <- data.frame(response = predict(object, type = "response", ...)) 
    rval$prob <- predict(object, type = "prob", ...) 

    ## get rules for each node 
    rls <- partykit:::.list.rules.party(object) 

    ## get predicted node and select corresponding rule 
    rval$rule <- rls[as.character(predict(object, type = "node", ...))] 

    return(rval) 
} 

pred <- pathpred(model) 
pred[c(1, 51, 101), ] 

を使用しています。しかし、それは私がrpart

答えて