Combine TreeModel and RegTree (#3995)

This commit is contained in:
Rory Mitchell
2018-12-19 12:16:40 +13:00
committed by GitHub
parent c055a32609
commit 84c99f86f4
10 changed files with 559 additions and 605 deletions

View File

@@ -122,7 +122,6 @@ class QuantileHistMock : public QuantileHistMaker {
gmat.Init((*dmat).get(), max_bins);
RegTree tree = RegTree();
tree.InitModel();
tree.param.InitAllowUnknown(cfg);
std::vector<GradientPair> gpair =
@@ -134,7 +133,6 @@ class QuantileHistMock : public QuantileHistMaker {
void TestBuildHist() {
RegTree tree = RegTree();
tree.InitModel();
tree.param.InitAllowUnknown(cfg);
size_t constexpr max_bins = 4;
@@ -146,7 +144,6 @@ class QuantileHistMock : public QuantileHistMaker {
void TestEvaluateSplit() {
RegTree tree = RegTree();
tree.InitModel();
tree.param.InitAllowUnknown(cfg);
builder_->TestEvaluateSplit(gmatb_, tree);