Commit 3e3d76f6 authored by Joan Fontanals's avatar Joan Fontanals Committed by Nikita Titov
Browse files

[Java] MMLSPARK-ISSUE-720 - Create SWIG version of BoosterDumpModel that...

[Java] MMLSPARK-ISSUE-720 - Create SWIG version of BoosterDumpModel that returns string to be used in Java (#2560)
parent 7feaaefb
...@@ -53,6 +53,26 @@ ...@@ -53,6 +53,26 @@
return dst; return dst;
} }
char * LGBM_BoosterDumpModelSWIG(BoosterHandle handle,
int start_iteration,
int num_iteration,
int64_t buffer_len,
int64_t* out_len) {
char* dst = new char[buffer_len];
int result = LGBM_BoosterDumpModel(handle, start_iteration, num_iteration, buffer_len, out_len, dst);
// Reallocate to use larger length
if (*out_len > buffer_len) {
delete [] dst;
int64_t realloc_len = *out_len;
dst = new char[realloc_len];
result = LGBM_BoosterDumpModel(handle, start_iteration, num_iteration, realloc_len, out_len, dst);
}
if (result != 0) {
return nullptr;
}
return dst;
}
char ** LGBM_BoosterGetEvalNamesSWIG(BoosterHandle handle, char ** LGBM_BoosterGetEvalNamesSWIG(BoosterHandle handle,
int eval_counts) { int eval_counts) {
char** dst = new char*[eval_counts]; char** dst = new char*[eval_counts];
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment