added XGDMatrixNumCol_R function

2016-03-27 19:11:22 -05:00
parent 605c23e0dc
commit 1d504d6c6c
2 changed files with 15 additions and 1 deletions
--- a/R-package/src/xgboost_R.cc
+++ b/R-package/src/xgboost_R.cc
@@ -195,6 +195,14 @@ SEXP XGDMatrixNumRow_R(SEXP handle) {
  return ScalarInteger(static_cast<int>(nrow));
 }

+SEXP XGDMatrixNumCol_R(SEXP handle) {
+  bst_ulong ncol;
+  R_API_BEGIN();
+  CHECK_CALL(XGDMatrixNumCol(R_ExternalPtrAddr(handle), &ncol));
+  R_API_END();
+  return ScalarInteger(static_cast<int>(ncol));
+}
+
 // functions related to booster
 void _BoosterFinalizer(SEXP ext) {
  if (R_ExternalPtrAddr(ext) == NULL) return;