Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update xgboost.R

Browse Source 
add parameter missing
This commit is contained in:
Tong He 2014-11-20 15:14:05 -08:00
parent 20817b56f3
commit 98ee7e8057
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
R-package/R/xgboost.R
View File

@ -43,9 +43,14 @@
#'
#' @export
#'
xgboost <- function(data = NULL, label = NULL, params = list(), nrounds,
xgboost <- function(data = NULL, label = NULL, missing = NULL, params = list(), nrounds,
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(...))
if (verbose > 0) {