Skip to content

Instantly share code, notes, and snippets.

@szilard
Created October 14, 2016 04:44
Show Gist options
  • Save szilard/f80aa96d009a7c0e8452089a898af4d0 to your computer and use it in GitHub Desktop.
Save szilard/f80aa96d009a7c0e8452089a898af4d0 to your computer and use it in GitHub Desktop.
H2O Steam deploy GBM
library(h2o)
h2o.init(nthreads = -1)
dx_train <- h2o.importFile("https://s3.amazonaws.com/benchm-ml--main/train-1m.csv")
system.time({
md_10 <- h2o.gbm(x = 1:(ncol(dx_train)-1), y = ncol(dx_train), training_frame = dx_train,
model_id = "airline_depth10",
ntrees = 100, max_depth = 10, learn_rate = 0.1, nbins = 100)
})
system.time({
md_15 <- h2o.gbm(x = 1:(ncol(dx_train)-1), y = ncol(dx_train), training_frame = dx_train,
model_id = "airline_depth15",
ntrees = 100, max_depth = 15, learn_rate = 0.1, nbins = 100)
})
system.time({
md_20 <- h2o.gbm(x = 1:(ncol(dx_train)-1), y = ncol(dx_train), training_frame = dx_train,
model_id = "airline_depth20",
ntrees = 100, max_depth = 20, learn_rate = 0.1, nbins = 100)
})
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment