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

fix a tiny bug in xgboost

Browse Source
This commit is contained in:
Tong He 2014-08-27 15:51:34 -07:00
parent b151617ac1
commit 114cfb2167
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,7 +6,7 @@ xgboost <- function(data = NULL, label = NULL, params = list(), nrounds = 10,
if (inClass == "dgCMatrix" || inClass == "matrix") {
if (is.null(label))
stop("xgboost: need label when data is a matrix")
dtrain <- xgb.DMatrix(data, label = y)
dtrain <- xgb.DMatrix(data, label = label)
} else {
if (!is.null(label))
warning("xgboost: label will be ignored.")