Skip to content

GitLab

Unverified Commit 60710c70 authored by Alberto Ferreira's avatar Alberto Ferreira Committed by GitHub
Browse files

Fix SingleRowPredictor::IsPredictorEqual comparison (invert) (#2799)

parent 060c681d
Show whitespace changes
Inline Side-by-side
Showing with 7 additions and 5 deletions
src/c_api.cpp
View file @ 60710c70
...@@ -77,13 +77,15 @@ class SingleRowPredictor { ...@@ -77,13 +77,15 @@ class SingleRowPredictor {
predict_function = predictor_->GetPredictFunction(); predict_function = predictor_->GetPredictFunction();
num_total_model_ = boosting->NumberOfTotalModel(); num_total_model_ = boosting->NumberOfTotalModel();
} }
~SingleRowPredictor() {} ~SingleRowPredictor() {}
bool IsPredictorEqual(const Config& config, int iter, Boosting* boosting) { bool IsPredictorEqual(const Config& config, int iter, Boosting* boosting) {
return early_stop_ != config.pred_early_stop || return early_stop_ == config.pred_early_stop &&
early_stop_freq_ != config.pred_early_stop_freq || early_stop_freq_ == config.pred_early_stop_freq &&
early_stop_margin_ != config.pred_early_stop_margin || early_stop_margin_ == config.pred_early_stop_margin &&
iter_ != iter || iter_ == iter &&
num_total_model_ != boosting->NumberOfTotalModel(); num_total_model_ == boosting->NumberOfTotalModel();
} }
private: private:
......
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