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

Fix multi-output with alternating strategies. (#9933)

Browse Source 
---------

Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
This commit is contained in:
Jiaming Yuan
2024-01-04 16:41:13 +08:00
committed by GitHub
parent 5f7b5a6921
commit 621348abb3
6 changed files with 123 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/tree/updater_quantile_hist.cc
View File

@@ -545,12 +545,12 @@ class QuantileHistMaker : public TreeUpdater {
}
bool UpdatePredictionCache(const DMatrix *data, linalg::MatrixView<float> out_preds) override {
if (p_impl_) {
return p_impl_->UpdatePredictionCache(data, out_preds);
} else if (p_mtimpl_) {
if (out_preds.Shape(1) > 1) {
CHECK(p_mtimpl_);
return p_mtimpl_->UpdatePredictionCache(data, out_preds);
} else {
return false;
CHECK(p_impl_);
return p_impl_->UpdatePredictionCache(data, out_preds);
}
}