update LightGBM SWIG wrapper (#1610)

08c82ee6 · Ilya Matiach · Guolin Ke · 08add07b · 08c82ee6
Commit 08c82ee6 authored Aug 25, 2018 by Ilya Matiach Committed by Guolin Ke Aug 25, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

swig/lightgbmlib.i swig/lightgbmlib.i +2 -1

No files found.
--- a/swig/lightgbmlib.i
+++ b/swig/lightgbmlib.i
@@ -16,11 +16,12 @@

 %inline %{
  char * LGBM_BoosterSaveModelToStringSWIG(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_BoosterSaveModelToString(handle, num_iteration, buffer_len, out_len, dst);
+    int result = LGBM_BoosterSaveModelToString(handle, start_iteration, num_iteration, buffer_len, out_len, dst);
    if (result != 0) {
      return nullptr;
    }