Advertisement
Guest User

Untitled

a guest
Feb 6th, 2016
55
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.33 KB | None | 0 0
  1. library(xgboost)
  2.  
  3. X = as.matrix(train[,-58])
  4. y = train[,58]
  5. numberOfClasses <- max(y) + 1
  6.  
  7. bst <- xgboost(data = X, label = y, max.depth = 2, eta = 1, nthread = 2, nround = 200, objective = "binary:logistic", verbose = 1)
  8.  
  9. pred = predict(bst, data.matrix(test[,-58]))
  10. pred = as.numeric(pred > 0.5)
  11. mean(as.numeric(pred > 0.5) != test$spam)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement