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

Use view for SparsePage exclusively. (#6590)

Browse Source
This commit is contained in:
Jiaming Yuan
2021-01-11 18:04:55 +08:00
committed by GitHub
parent 78f2cd83d7
commit f2f7dd87b8
23 changed files with 151 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/cpp/c_api/test_c_api.cc
View File

@@ -30,10 +30,11 @@ TEST(CAPI, XGDMatrixCreateFromMatDT) {
ASSERT_EQ(info.num_nonzero_, 6ul);
for (const auto &batch : (*dmat)->GetBatches<xgboost::SparsePage>()) {
ASSERT_EQ(batch[0][0].fvalue, 0.0f);
ASSERT_EQ(batch[0][1].fvalue, -4.0f);
ASSERT_EQ(batch[2][0].fvalue, 3.0f);
ASSERT_EQ(batch[2][1].fvalue, 0.0f);
auto page = batch.GetView();
ASSERT_EQ(page[0][0].fvalue, 0.0f);
ASSERT_EQ(page[0][1].fvalue, -4.0f);
ASSERT_EQ(page[2][0].fvalue, 3.0f);
ASSERT_EQ(page[2][1].fvalue, 0.0f);
}
delete dmat;
@@ -62,8 +63,9 @@ TEST(CAPI, XGDMatrixCreateFromMatOmp) {
ASSERT_EQ(info.num_nonzero_, num_cols * row - num_missing);
for (const auto &batch : (*dmat)->GetBatches<xgboost::SparsePage>()) {
auto page = batch.GetView();
for (size_t i = 0; i < batch.Size(); i++) {
auto inst = batch[i];
auto inst = page[i];
for (auto e : inst) {
ASSERT_EQ(e.fvalue, 1.5);
}