Guest User

Untitled

a guest
Jan 23rd, 2018
91
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.39 KB | None | 0 0
  1. library (rpart)
  2. library(partykit)
  3.  
  4. data("Titanic", package = "datasets")
  5. ttnc <- as.data.frame(Titanic)
  6. ttnc <- ttnc[rep(1:nrow(ttnc), ttnc$Freq), 1:4]
  7. names(ttnc)[2] <- "Gender"
  8.  
  9. rp <- rpart(Survived ~ Gender + Age + Class, data = ttnc)
  10. prp <- as.party(rp)
  11.  
  12. prp[5]
  13.  
  14. data_party(prp, id = 5)
  15.  
  16. ttnc$Node <- predict(prp, newdata = ttnc, type = "node")
  17.  
  18. subtree <- subset(ttnc, Node == 5)
Add Comment
Please, Sign In to add comment