[CPP] disable early stopping in dart (#1894)

* [CPP] disable early stopping in dart * fix compile error
microsoft · Dec 10, 2018 · 3f1fc83 · 3f1fc83
1 parent 47f0d5e
commit 3f1fc83
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/boosting/dart.hpp b/src/boosting/dart.hpp
@@ -78,6 +78,11 @@ class DART: public GBDT {
     *out_len = static_cast<int64_t>(train_score_updater_->num_data()) * num_class_;
     return train_score_updater_->score();
   }
+
+  bool EvalAndCheckEarlyStopping() override {
+    GBDT::OutputMetric(iter_);
+    return false;
+  }
 
 private:
   /*!

diff --git a/src/boosting/gbdt.h b/src/boosting/gbdt.h
@@ -360,7 +360,7 @@ class GBDT : public GBDTBase {
   /*!
   * \brief Print eval result and check early stopping
   */
-  bool EvalAndCheckEarlyStopping();
+  virtual bool EvalAndCheckEarlyStopping();
 
   /*!
   * \brief reset config for bagging