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

[EM] Support SHAP contribution with QDM. (#10724)

Browse Source 
- Add GPU support.
- Add external memory support.
- Update the GPU tree shap.
This commit is contained in:
Jiaming Yuan
2024-08-22 05:25:10 +08:00
committed by GitHub
parent cb54374550
commit 142bdc73ec
13 changed files with 274 additions and 159 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/cpp/data/test_simple_dmatrix.cc
View File

@@ -1,5 +1,5 @@
/**
* Copyright 2016-2023 by XGBoost Contributors
* Copyright 2016-2024, XGBoost Contributors
*/
#include <xgboost/data.h>
@@ -434,12 +434,11 @@ namespace {
void VerifyColumnSplit() {
size_t constexpr kRows {16};
size_t constexpr kCols {8};
auto dmat =
RandomDataGenerator{kRows, kCols, 0}.GenerateDMatrix(false, false, 1, DataSplitMode::kCol);
auto p_fmat = RandomDataGenerator{kRows, kCols, 0}.GenerateDMatrix(false, DataSplitMode::kCol);
ASSERT_EQ(dmat->Info().num_col_, kCols * collective::GetWorldSize());
ASSERT_EQ(dmat->Info().num_row_, kRows);
ASSERT_EQ(dmat->Info().data_split_mode, DataSplitMode::kCol);
ASSERT_EQ(p_fmat->Info().num_col_, kCols * collective::GetWorldSize());
ASSERT_EQ(p_fmat->Info().num_row_, kRows);
ASSERT_EQ(p_fmat->Info().data_split_mode, DataSplitMode::kCol);
}
} // anonymous namespace