fixed a bug in model output

a194045c · Guolin Ke · GitHub · 5d022898 · a194045c
Commit a194045c authored Oct 31, 2016 by Guolin Ke Committed by GitHub Oct 31, 2016
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 5 deletions

src/boosting/gbdt.cpp src/boosting/gbdt.cpp +3 -5

No files found.
--- a/src/boosting/gbdt.cpp
+++ b/src/boosting/gbdt.cpp
@@ -317,11 +317,9 @@ void GBDT::SaveModelToFile(bool is_finish, const char* filename) {
    model_output_file_ << "Tree=" << i << std::endl;
    model_output_file_ << models_[i]->ToString() << std::endl;
  }
-
-  if (rest > 0) {
-    saved_model_size_ = rest;
-  }
-
+  
+  saved_model_size_ = Common::Max(saved_model_size_, rest);
+  
  model_output_file_.flush();
  // training finished, can close file
  if (is_finish) {