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

Fix length check in utils.R

Browse Source
This commit is contained in:
nagadomi 2015-04-09 01:20:43 +09:00
parent e91bacd378
commit 87b4332cc1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
R-package/R/utils.R
View File

@ -36,8 +36,8 @@ xgb.setinfo <- function(dmat, name, info) {
return(TRUE)
}
if (name == "group") {
if (length(info)!=xgb.numrow(dmat))
stop("The length of groups must equal to the number of rows in the input data")
if (sum(info)!=xgb.numrow(dmat))
stop("The sum of groups must equal to the number of rows in the input data")
.Call("XGDMatrixSetInfo_R", dmat, name, as.integer(info),
PACKAGE = "xgboost")
return(TRUE)