Update xgboost.R
add parameter missing
This commit is contained in:
parent
20817b56f3
commit
98ee7e8057
@ -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) {
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user