#
# Example R code for generating an H2O Scoring POJO.
#

# "Safe" system.  Error checks process exit status code.  stop() if it failed.
safeSystem <- function(x) {
  print(sprintf("+ CMD: %s", x))
  res <- system(x)
  print(res)
  if (res != 0) {
    msg <- sprintf("SYSTEM COMMAND FAILED (exit status %d)", res)
    stop(msg)
  }
}

library(h2o)

cat("Starting H2O\n")
myIP <- "localhost"
myPort <- 54321
h <- h2o.init(ip = myIP, port = myPort, startH2O = TRUE)

cat("Building GBM model\n")
df <- h2o.importFile(path = normalizePath("./training_data.csv"));
y <- "Label"
x <- c("Has4Legs","CoatColor","HairLength","TailLength","EnjoysPlay","StaresOutWindow","HoursSpentNapping","RespondsToCommands","EasilyFrightened","Age", "Noise1", "Noise2", "Noise3", "Noise4", "Noise5")
gbm.h2o.fit <- h2o.gbm(training_frame = df, y = y, x = x, model_id = "GBMPojo", ntrees = 10)

cat("Downloading Java prediction model code from H2O\n")
model_id <- gbm.h2o.fit@model_id

tmpdir_name <- "generated_model"
cmd <- sprintf("rm -fr %s", tmpdir_name)
safeSystem(cmd)
cmd <- sprintf("mkdir %s", tmpdir_name)
safeSystem(cmd)

h2o.download_pojo(gbm.h2o.fit, "./generated_model/")

cat("Note: H2O will shut down automatically if it was started by this R script and the script exits\n")