MCPcopy Create free account
hub / github.com/antmachineintelligence/mtgbmcode / RollbackOneIter

Method RollbackOneIter

src/boosting/gbdt.cpp:550–566  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

548}
549
550void GBDT::RollbackOneIter() {
551 if (iter_ <= 0) { return; }
552 // reset score
553 for (int cur_tree_id = 0; cur_tree_id < num_tree_per_iteration_; ++cur_tree_id) {
554 auto curr_tree = models_.size() - num_tree_per_iteration_ + cur_tree_id;
555 models_[curr_tree]->Shrinkage(-1.0);
556 train_score_updater_->AddScore(models_[curr_tree].get(), cur_tree_id);
557 for (auto& score_updater : valid_score_updater_) {
558 score_updater->AddScore(models_[curr_tree].get(), cur_tree_id);
559 }
560 }
561 // remove model
562 for (int cur_tree_id = 0; cur_tree_id < num_tree_per_iteration_; ++cur_tree_id) {
563 models_.pop_back();
564 }
565 --iter_;
566}
567
568bool GBDT::EvalAndCheckEarlyStopping() {
569 bool is_met_early_stopping = false;

Callers

nothing calls this directly

Calls 5

pop_backMethod · 0.80
sizeMethod · 0.45
ShrinkageMethod · 0.45
AddScoreMethod · 0.45
getMethod · 0.45

Tested by

no test coverage detected