Update xgboost.R

add parameter missing
This commit is contained in:
Tong He 2014-11-20 15:14:05 -08:00
parent 20817b56f3
commit 98ee7e8057

View File

@ -43,9 +43,14 @@
#' #'
#' @export #' @export
#' #'
xgboost <- function(data = NULL, label = NULL, params = list(), nrounds, xgboost <- function(data = NULL, label = NULL, missing = NULL, params = list(), nrounds,
verbose = 1, ...) { verbose = 1, ...) {
dtrain <- xgb.get.DMatrix(data, label) if (is.null(missing)) {
dtrain <- xgb.get.DMatrix(data, label)
} else {
dtrain <- xgb.get.DMatrix(data, label, missing)
}
params <- append(params, list(...)) params <- append(params, list(...))
if (verbose > 0) { if (verbose > 0) {