我做了一个决策树模型。 我获得了模型的召回率和精度。但是,我需要获取每个节点的召回率和精度。
如何用R做到这一点?
我的代码:
tree1<- rpart(Interested ~ . , data= training_set, method = 'class', control =rpart.control(minsplit =16, minbucket =16 , cp=0))
rpart.plot(tree1, extra = 4) # pic 1.2
printcp(tree1)
plotcp(tree1) #min error : 0.000219
ptree1 <- prune(tree1, cp=0.000219)
rpart.plot(ptree1, extra = 4, uniform=TRUE) #--trining
pred=predict(tree1, training_set, type='class')
table(pred = pred, true=training_set$Interested)