Guest User

Untitled

a guest
Nov 24th, 2017
104
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.54 KB | None | 0 0
  1. set.seed(15)
  2. train.data <- final_data_Fpositive[sample(358,215),]
  3. set.seed(15)
  4. test.data <- final_data_Fpositive[-sample(358,215),]
  5.  
  6. CLtree <- rpart(TB~., data = train.data, method = "class")
  7. plot(CLtree)
  8.  
  9. text(CLtree, pretty = 0)
  10.  
  11. printcp(CLtree)
  12. plotcp(CLtree)
  13.  
  14. ptree<- prune(CLtree, cp= CLtree$cptable[which.min(CLtree$cptable[,"xerror"]),"CP"])
  15.  
  16. plot(ptree, uniform=TRUE, main="Pruned Classification Tree")
  17. text(ptree, cex = 0.7)
  18.  
  19. pred.pruned <- predict(ptree, test.data, "class")
  20. mean(pred.pruned == test.data$TB)
  21. table(test.data$TB,pred.pruned)
Add Comment
Please, Sign In to add comment