xgboost_jug.R

library(jug)
library(xgboost)
library(Matrix)

setwd("~/Projects/xgboost_test")
model <- xgb.load("xgboost1.save")

jug() %>%
  get("/model/(?<age>.*)/(?<workclass>.*)/(?<education>.*)/(?<marital_status>.*)/(?<occupation>.*)/(?<relationship>.*)/(?<gender>.*)", function(req, res, err){
    df <- data.frame(
      age = as.numeric(req$params$age),
      workclass = req$params$workclass,
      education = req$params$education,
      marital_status = req$params$marital_status,
      occupation = req$params$occupation,
      relationship = req$params$relationship,
      gender = req$params$gender
    )
    df$score <- predict(model, data.matrix(df))
    df$prediction <- df$score>0.5
    cat(str(df))
    res$json(df)
  }) %>%
  simple_error_handler_json() %>%
  serve_it()