Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

reset learner

Browse Source
This commit is contained in:
Hendrik Groove 2024-10-21 23:49:22 +02:00
parent 1931a70598
commit 1c666db349
1 changed files with 19 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/learner.cc
View File

@ -61,8 +61,6 @@
#include "xgboost/predictor.h" // for PredictionContainer, PredictionCacheEntry #include "xgboost/predictor.h" // for PredictionContainer, PredictionCacheEntry
#include "xgboost/string_view.h" // for operator<<, StringView #include "xgboost/string_view.h" // for operator<<, StringView
#include "xgboost/task.h" // for ObjInfo #include "xgboost/task.h" // for ObjInfo
#include <iostream>
#include <exception>
namespace { namespace {
const char* kMaxDeltaStepDefaultValue = "0.7"; const char* kMaxDeltaStepDefaultValue = "0.7";
@ -1264,12 +1262,10 @@ class LearnerImpl : public LearnerIO {
return out_impl; return out_impl;
} }
void UpdateOneIter(int iter, std::shared_ptr<DMatrix> train) override { void UpdateOneIter(int iter, std::shared_ptr<DMatrix> train) override {
monitor_.Start("UpdateOneIter"); monitor_.Start("UpdateOneIter");
TrainingObserver::Instance().Update(iter); TrainingObserver::Instance().Update(iter);
this->Configure(); this->Configure();
this->InitBaseScore(train.get()); this->InitBaseScore(train.get());
if (ctx_.seed_per_iteration) { if (ctx_.seed_per_iteration) {
@ -1286,22 +1282,13 @@ void UpdateOneIter(int iter, std::shared_ptr<DMatrix> train) override {
monitor_.Stop("PredictRaw"); monitor_.Stop("PredictRaw");
monitor_.Start("GetGradient"); monitor_.Start("GetGradient");
try {
GetGradient(predt.predictions, train->Info(), iter, &gpair_); GetGradient(predt.predictions, train->Info(), iter, &gpair_);
} catch (const std::exception& e) {
throw;
}
monitor_.Stop("GetGradient"); monitor_.Stop("GetGradient");
TrainingObserver::Instance().Observe(*gpair_.Data(), "Gradients"); TrainingObserver::Instance().Observe(*gpair_.Data(), "Gradients");
try {
gbm_->DoBoost(train.get(), &gpair_, &predt, obj_.get()); gbm_->DoBoost(train.get(), &gpair_, &predt, obj_.get());
} catch (const std::exception& e) {
throw;
}
monitor_.Stop("UpdateOneIter"); monitor_.Stop("UpdateOneIter");
} }
void BoostOneIter(int iter, std::shared_ptr<DMatrix> train, void BoostOneIter(int iter, std::shared_ptr<DMatrix> train,
linalg::Matrix<GradientPair>* in_gpair) override { linalg::Matrix<GradientPair>* in_gpair) override {