Guest User

Untitled

a guest
Dec 16th, 2017
152
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.50 KB | None | 0 0
  1. train = paint_train %>% select(-logprice)
  2. train.y = paint_train %>% select(logprice) %>% pull()
  3. data.new <- paint_test %>% select(-logprice) %>% data.matrix()
  4. xgb.fit <- xgboost(data = data.matrix(train),
  5. label = train.y,
  6. objective = "reg:linear",
  7. eval_metric = "rmse",
  8. max.depth = 10,
  9. eta = 0.05,
  10. nround = 75,
  11. verbose =0)
  12.  
  13. xgb.pred <- exp(predict(xgb.fit, newdata = data.new))
Add Comment
Please, Sign In to add comment