| 18 | |
| 19 | namespace xgboost { |
| 20 | inline std::unique_ptr<gbm::GBTreeModel> CreateTestModel(LearnerModelParam const* param, |
| 21 | Context const* ctx, size_t n_classes = 1) { |
| 22 | auto model = std::make_unique<gbm::GBTreeModel>(param, ctx); |
| 23 | |
| 24 | for (size_t i = 0; i < n_classes; ++i) { |
| 25 | std::vector<std::unique_ptr<RegTree>> trees; |
| 26 | trees.push_back(std::unique_ptr<RegTree>(new RegTree)); |
| 27 | if (i == 0) { |
| 28 | (*trees.back())[0].SetLeaf(1.5f); |
| 29 | (*trees.back()).Stat(0).sum_hess = 1.0f; |
| 30 | } |
| 31 | model->CommitModelGroup(std::move(trees), i); |
| 32 | } |
| 33 | |
| 34 | return model; |
| 35 | } |
| 36 | |
| 37 | inline auto CreatePredictorForTest(Context const* ctx) { |
| 38 | if (ctx->IsCPU()) { |
no test coverage detected