Sign Up to our social questions and Answers Engine to ask questions, answer people’s questions, and connect with other people.
Login to our social questions & Answers Engine to ask questions answer people’s questions & connect with other people.
Lost your password? Please enter your email address. You will receive a link and will create a new password via email.
Please briefly explain why you feel this question should be reported.
Please briefly explain why you feel this answer should be reported.
Please briefly explain why you feel this user should be reported.
I’m using R’s randomForest package. Is there a way for me to figure out the splitting rules used at each node in the computed trees?
Sample code:
library(randomForest)
mydata = data.frame(output = factor(c(0, 0, 0, 1, 1, 1)), x = c(0, 1, 0, 0, 1, 1), y = c(1, 1, 1, 0, 0, 1))
mydata.rf = randomForest(output ~ ., data = mydata, ntree = 3)
I suspect it has something to do with mydata.rf$forest$treemap, but I’m not sure.
Ah, answered it myself: I can use the
getTreefunction, described in the pdf documentation.