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

Implement GPU predict leaf. (#6187)

Browse Source
This commit is contained in:
Jiaming Yuan
2020-11-11 17:33:47 +08:00
committed by GitHub
parent 7f101d1b33
commit 8a17610666
12 changed files with 252 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/gbm/gblinear.cc
View File

@@ -147,9 +147,7 @@ class GBLinear : public GradientBooster {
}
}
void PredictLeaf(DMatrix*,
std::vector<bst_float>*,
unsigned) override {
void PredictLeaf(DMatrix *, HostDeviceVector<bst_float> *, unsigned) override {
LOG(FATAL) << "gblinear does not support prediction of leaf index";
}

5
src/gbm/gbtree.h
View File

@@ -278,10 +278,9 @@ class GBTree : public GradientBooster {
}
void PredictLeaf(DMatrix* p_fmat,
std::vector<bst_float>* out_preds,
HostDeviceVector<bst_float>* out_preds,
unsigned ntree_limit) override {
CHECK(configured_);
cpu_predictor_->PredictLeaf(p_fmat, out_preds, model_, ntree_limit);
this->GetPredictor()->PredictLeaf(p_fmat, out_preds, model_, ntree_limit);
}
void PredictContribution(DMatrix* p_fmat,