Advertisement
Guest User

Untitled

a guest
Jun 29th, 2016
70
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.42 KB | None | 0 0
  1. dat <- data.frame(y = runif(1000), x=rnorm(1000))
  2.  
  3. gbmMod <- gbm::gbm(y~x, data=dat, n.trees=5000, cv.folds=0)
  4.  
  5. summary(lm(predict(gbmMod, n.trees=5000) ~ dat$y))$adj.r.squared
  6.  
  7. inds <- sample(1:nrow(dat), 0.7*nrow(dat))
  8.  
  9. train <- dat[inds, ]
  10. test <- dat[-inds, ]
  11.  
  12. gbmMod2 <- gbm::gbm(y~x, data=train, n.trees=5000)
  13.  
  14. preds <- predict(gbmMod2, newdata = test, n.trees=5000)
  15.  
  16. summary(lm(preds ~ test[,1]))$adj.r.squared
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement