Skip to content

GitLab

Commit a194045c authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

fixed a bug in model output

parent 5d022898
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 5 deletions
src/boosting/gbdt.cpp
View file @ a194045c
...@@ -317,11 +317,9 @@ void GBDT::SaveModelToFile(bool is_finish, const char* filename) { ...@@ -317,11 +317,9 @@ void GBDT::SaveModelToFile(bool is_finish, const char* filename) {
model_output_file_ << "Tree=" << i << std::endl; model_output_file_ << "Tree=" << i << std::endl;
model_output_file_ << models_[i]->ToString() << std::endl; model_output_file_ << models_[i]->ToString() << std::endl;
} }
if (rest > 0) { saved_model_size_ = Common::Max(saved_model_size_, rest);
saved_model_size_ = rest;
}
model_output_file_.flush(); model_output_file_.flush();
// training finished, can close file // training finished, can close file
if (is_finish) { if (is_finish) {
......
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