MCPcopy Create free account
hub / github.com/dmlc/xgboost / TestInitAfterLoad

Method TestInitAfterLoad

tests/cpp/test_learner.cc:541–567  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

539 }
540
541 void TestInitAfterLoad() {
542 std::unique_ptr<Learner> learner{Learner::Create({Xy_})};
543 learner->SetParam("objective", "reg:absoluteerror");
544 learner->Configure();
545
546 Json model{Object{}};
547 learner->SaveModel(&model);
548 auto base_score = GetBaseScore(model);
549 ASSERT_EQ(base_score.size(), 1);
550 ASSERT_FALSE(std::isnan(base_score[0]));
551 ASSERT_EQ(base_score[0], ObjFunction::DefaultBaseScore());
552
553 learner.reset(Learner::Create({Xy_}));
554 learner->LoadModel(model);
555 Json config(Object{});
556 learner->Configure();
557 learner->SaveConfig(&config);
558 base_score = GetBaseScore(config);
559 ASSERT_EQ(base_score[0], ObjFunction::DefaultBaseScore());
560
561 learner->UpdateOneIter(0, Xy_);
562 learner->SaveConfig(&config);
563 base_score = GetBaseScore(config);
564 ASSERT_EQ(base_score.size(), 1);
565 ASSERT_FALSE(std::isnan(base_score[0]));
566 ASSERT_NE(base_score[0], ObjFunction::DefaultBaseScore());
567 }
568
569 void TestInitWithPredt() {
570 std::unique_ptr<Learner> learner{Learner::Create({Xy_})};

Callers 1

TEST_FFunction · 0.80

Calls 9

GetBaseScoreFunction · 0.85
SetParamMethod · 0.80
UpdateOneIterMethod · 0.80
ConfigureMethod · 0.45
SaveModelMethod · 0.45
sizeMethod · 0.45
resetMethod · 0.45
LoadModelMethod · 0.45
SaveConfigMethod · 0.45

Tested by

no test coverage detected