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

Combine TreeModel and RegTree (#3995)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/cpp/predictor/test_cpu_predictor.cc
View File

@@ -10,7 +10,6 @@ TEST(cpu_predictor, Test) {
std::vector<std::unique_ptr<RegTree>> trees;
trees.push_back(std::unique_ptr<RegTree>(new RegTree));
trees.back()->InitModel();
(*trees.back())[0].SetLeaf(1.5f);
(*trees.back()).Stat(0).sum_hess = 1.0f;
gbm::GBTreeModel model(0.5);

2
tests/cpp/predictor/test_gpu_predictor.cu
View File

@@ -35,7 +35,6 @@ TEST(gpu_predictor, Test) {
std::vector<std::unique_ptr<RegTree>> trees;
trees.push_back(std::unique_ptr<RegTree>(new RegTree()));
trees.back()->InitModel();
(*trees.back())[0].SetLeaf(1.5f);
(*trees.back()).Stat(0).sum_hess = 1.0f;
gbm::GBTreeModel model(0.5);
@@ -181,7 +180,6 @@ TEST(gpu_predictor, MGPU_Test) {
std::vector<std::unique_ptr<RegTree>> trees;
trees.push_back(std::unique_ptr<RegTree>(new RegTree()));
trees.back()->InitModel();
(*trees.back())[0].SetLeaf(1.5f);
(*trees.back()).Stat(0).sum_hess = 1.0f;
gbm::GBTreeModel model(0.5);