Skip to content

GitLab

Unverified Commit 3f1fc835 authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

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

* [CPP] disable early stopping in dart

* fix compile error
parent 47f0d5e8
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 1 deletion
src/boosting/dart.hpp
View file @ 3f1fc835
...@@ -78,6 +78,11 @@ public: ...@@ -78,6 +78,11 @@ public:
*out_len = static_cast<int64_t>(train_score_updater_->num_data()) * num_class_; *out_len = static_cast<int64_t>(train_score_updater_->num_data()) * num_class_;
return train_score_updater_->score(); return train_score_updater_->score();
} }
bool EvalAndCheckEarlyStopping() override {
GBDT::OutputMetric(iter_);
return false;
}
private: private:
/*! /*!
......
src/boosting/gbdt.h
View file @ 3f1fc835
...@@ -360,7 +360,7 @@ protected: ...@@ -360,7 +360,7 @@ protected:
/*! /*!
* \brief Print eval result and check early stopping * \brief Print eval result and check early stopping
*/ */
bool EvalAndCheckEarlyStopping(); virtual bool EvalAndCheckEarlyStopping();
/*! /*!
* \brief reset config for bagging * \brief reset config for bagging
......
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