f05a23b41c
- Fix case where Python reuses id from freed objects. - Small optimization to column matrix with QDM by using `realloc` instead of copying data.
41 lines
1.0 KiB
C++
41 lines
1.0 KiB
C++
/**
|
|
* Copyright 2022-2023 by XGBoost contributors
|
|
*/
|
|
#include "test_iterative_dmatrix.h"
|
|
|
|
#include <gtest/gtest.h>
|
|
#include <limits>
|
|
#include <memory>
|
|
|
|
#include "../../../src/data/gradient_index.h"
|
|
#include "../../../src/data/iterative_dmatrix.h"
|
|
#include "../helpers.h"
|
|
#include "xgboost/data.h" // DMatrix
|
|
|
|
namespace xgboost::data {
|
|
TEST(IterativeDMatrix, Ref) {
|
|
Context ctx;
|
|
TestRefDMatrix<GHistIndexMatrix, NumpyArrayIterForTest>(
|
|
&ctx, [&](GHistIndexMatrix const& page) { return page.cut; });
|
|
}
|
|
|
|
TEST(IterativeDMatrix, IsDense) {
|
|
bst_bin_t n_bins = 16;
|
|
auto test = [n_bins](float sparsity) {
|
|
NumpyArrayIterForTest iter(sparsity);
|
|
auto n_threads = 0;
|
|
IterativeDMatrix m(&iter, iter.Proxy(), nullptr, Reset, Next,
|
|
std::numeric_limits<float>::quiet_NaN(), n_threads, n_bins);
|
|
ASSERT_EQ(m.Ctx()->Threads(), AllThreadsForTest());
|
|
if (sparsity == 0.0) {
|
|
ASSERT_TRUE(m.IsDense());
|
|
} else {
|
|
ASSERT_FALSE(m.IsDense());
|
|
}
|
|
};
|
|
test(0.0);
|
|
test(0.1);
|
|
test(1.0);
|
|
}
|
|
} // namespace xgboost::data
|