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

* [CPP] disable early stopping in dart * fix compile error

[CPP] disable early stopping in dart (#1894)
* [CPP] disable early stopping in dart * fix compile error
3f1fc835 · Guolin Ke · GitHub · 47f0d5e8 · 3f1fc835 · 3f1fc835
Unverified Commit 3f1fc835 authored Dec 10, 2018 by Guolin Ke Committed by GitHub Dec 10, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

src/boosting/dart.hpp src/boosting/dart.hpp +5 -0

src/boosting/gbdt.h src/boosting/gbdt.h +1 -1

No files found.
--- a/src/boosting/dart.hpp
+++ b/src/boosting/dart.hpp
@@ -78,6 +78,11 @@ public:
    *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:
  /*!

--- a/src/boosting/gbdt.h
+++ b/src/boosting/gbdt.h
@@ -360,7 +360,7 @@ protected:
  /*!
  * \brief Print eval result and check early stopping
  */
-  bool EvalAndCheckEarlyStopping();
+  virtual bool EvalAndCheckEarlyStopping();

  /*!
  * \brief reset config for bagging