Update utils.R

This commit is contained in:
Tong He 2015-03-20 11:06:01 -07:00
parent 360cc7118d
commit f6722ba628

View File

@ -29,8 +29,8 @@ xgb.setinfo <- function(dmat, name, info) {
return(TRUE) return(TRUE)
} }
if (name == "base_margin") { if (name == "base_margin") {
if (length(info)!=xgb.numrow(dmat)) # if (length(info)!=xgb.numrow(dmat))
stop("The length of base margin must equal to the number of rows in the input data") # stop("The length of base margin must equal to the number of rows in the input data")
.Call("XGDMatrixSetInfo_R", dmat, name, as.numeric(info), .Call("XGDMatrixSetInfo_R", dmat, name, as.numeric(info),
PACKAGE = "xgboost") PACKAGE = "xgboost")
return(TRUE) return(TRUE)