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

Support column split in histogram builder (#8811)

Browse Source
This commit is contained in:
Rong Ou
2023-02-17 06:37:01 -08:00
committed by GitHub
parent 40fd3d6d5f
commit a65ad0bd9c
8 changed files with 38 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/predictor/cpu_predictor.cc
View File

@@ -584,7 +584,7 @@ class CPUPredictor : public Predictor {
void PredictDMatrix(DMatrix *p_fmat, std::vector<bst_float> *out_preds,
gbm::GBTreeModel const &model, int32_t tree_begin, int32_t tree_end) const {
if (p_fmat->Info().data_split_mode == DataSplitMode::kCol) {
if (p_fmat->IsColumnSplit()) {
ColumnSplitHelper helper(this->ctx_->Threads(), model, tree_begin, tree_end);
helper.PredictDMatrix(p_fmat, out_preds);
return;