This commit is contained in:
El Potaeto 2015-01-09 11:54:23 +01:00
parent d96bd15b7d
commit b656ca1554

View File

@ -276,9 +276,9 @@ extern "C" {
_WrapperBegin(); _WrapperBegin();
bst_ulong olen; bst_ulong olen;
const char **res = XGBoosterDumpModel(R_ExternalPtrAddr(handle), const char **res = XGBoosterDumpModel(R_ExternalPtrAddr(handle),
CHAR(asChar(fmap)), CHAR(asChar(fmap)),
asInteger(with_stats), asInteger(with_stats),
&olen); &olen);
SEXP out = PROTECT(allocVector(STRSXP, olen)); SEXP out = PROTECT(allocVector(STRSXP, olen));
char buffer [2000]; char buffer [2000];
for (size_t i = 0; i < olen; ++i) { for (size_t i = 0; i < olen; ++i) {
@ -287,7 +287,7 @@ extern "C" {
SET_STRING_ELT(out, i, mkChar(buffer)); SET_STRING_ELT(out, i, mkChar(buffer));
} }
_WrapperEnd(); _WrapperEnd();
UNPROTECT(1); UNPROTECT(1);
return out; return out;
} }
} }