具有R,召回率和精度的决策树模型

时间:2018-07-23 17:14:30

标签: decision-tree precision-recall

我做了一个决策树模型。 我获得了模型的召回率和精度。但是,我需要获取每个节点的召回率和精度。

如何用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)

0 个答案:

没有答案