add nrow to getinfo

This commit is contained in:
hetong 2015-01-19 13:36:53 -08:00
parent a1e188aa75
commit f295177b1d

View File

@ -39,7 +39,7 @@ setMethod("getinfo", signature = "xgb.DMatrix",
if (name != "nrow"){ if (name != "nrow"){
ret <- .Call("XGDMatrixGetInfo_R", object, name, PACKAGE = "xgboost") ret <- .Call("XGDMatrixGetInfo_R", object, name, PACKAGE = "xgboost")
} else { } else {
ret <- .Call("XGDMatrixNumRow_R", object) ret <- .Call("XGDMatrixNumRow_R", object, PACKAGE = "xgboost")
} }
return(ret) return(ret)
}) })