R/xgboost_func.R

Defines functions xgboost_wrapper

Documented in xgboost_wrapper

#' A wrapper for xgboost
#'
#' @param x numeric
#' @return xgboost trained
xgboost_wrapper <- function(x) {
  if (x == 1) {
    dtrain <- xgboost::xgb.DMatrix(data = as.matrix(datasets::iris[, 2:4]),
                          label = as.numeric(datasets::iris$Species))

    out <- xgboost::xgboost(data = dtrain,
                     max.depth = 2,
                     eta = 1,
                     nthread = 2,
                     nrounds = 2)
  }
  return(out)
}
jyuu/trytravis documentation built on Dec. 26, 2019, 7:39 a.m.