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
Show 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
@@ -318,9 +318,7 @@ void GBDT::SaveModelToFile(bool is_finish, const char* filename) {
    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