fix a silly bug.

b30beb99 · Guolin Ke · GitHub · 92f2a570 · b30beb99
Unverified Commit b30beb99 authored Dec 22, 2017 by Guolin Ke Committed by GitHub Dec 22, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/application/application.cpp src/application/application.cpp +1 -1

No files found.
--- a/src/application/application.cpp
+++ b/src/application/application.cpp
@@ -87,7 +87,7 @@ void Application::LoadData() {
  PredictFunction predict_fun = nullptr;
  PredictionEarlyStopInstance pred_early_stop = CreatePredictionEarlyStopInstance("none", LightGBM::PredictionEarlyStopConfig());
  // need to continue training
-  if (boosting_->NumberOfTotalModel() > 0 || config_.task_type != TaskType::KRefitTree) {
+  if (boosting_->NumberOfTotalModel() > 0 && config_.task_type != TaskType::KRefitTree) {
    predictor.reset(new Predictor(boosting_.get(), -1, true, false, false, false, -1, -1));
    predict_fun = predictor->GetPredictFunction();
  }