Move warning about empty dataset. (#5998)

2020-08-11 14:10:51 +08:00
parent f93f1c03fc
commit 6f7112a848
3 changed files with 4 additions and 6 deletions
--- a/src/learner.cc
+++ b/src/learner.cc
@@ -1119,6 +1119,10 @@ class LearnerImpl : public LearnerIO {
               "booster.";
      }
    }
+
+    if (p_fmat->Info().num_row_ == 0) {
+      LOG(WARNING) << "Empty dataset at worker: " << rabit::GetRank();
+    }
  }

 private: