From 6d6f964857624a3558da403277c05c24200d09af Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?E=2E=20Ari=C3=B1o=20de=20la=20Rubia?= Date: Mon, 29 May 2017 00:22:30 -0700 Subject: [PATCH] Create h2o.R for newly released h2o AutoML AUC of 0.7284624 for train-0.1m.csv --- automl/h2o.R | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 automl/h2o.R diff --git a/automl/h2o.R b/automl/h2o.R new file mode 100644 index 0000000..63655da --- /dev/null +++ b/automl/h2o.R @@ -0,0 +1,18 @@ +library(h2o) + +h2o.init(max_mem_size="60g", nthreads=-1) + +dx_train <- h2o.importFile(path = "train-0.1m.csv") +dx_test <- h2o.importFile(path = "test.csv") + +Xnames <- names(dx_train)[which(names(dx_train)!="dep_delayed_15min")] + +system.time({ + md <- h2o.automl(x = Xnames, y = "dep_delayed_15min", training_frame = dx_train) +}) + +system.time({ + print(h2o.auc(h2o.performance(md@leader, dx_test))) +}) + +# 0.7284624