|
17 | 17 | #include "xgboost/host_device_vector.h" // for HostDeviceVector
|
18 | 18 | #include "xgboost/json.h" // for Json, String, Object
|
19 | 19 |
|
20 |
| -namespace xgboost { |
21 |
| -namespace metric { |
| 20 | +namespace xgboost::metric { |
22 | 21 |
|
23 | 22 | inline void VerifyPrecision(DataSplitMode data_split_mode = DataSplitMode::kRow) {
|
24 |
| - // When the limit for precision is not given, it takes the limit at |
25 |
| - // std::numeric_limits<unsigned>::max(); hence all values are very small |
26 |
| - // NOTE(AbdealiJK): Maybe this should be fixed to be num_row by default. |
27 | 23 | auto ctx = xgboost::CreateEmptyGenericParam(GPUIDX);
|
28 |
| - xgboost::Metric * metric = xgboost::Metric::Create("pre", &ctx); |
| 24 | + xgboost::Metric* metric = xgboost::Metric::Create("pre", &ctx); |
29 | 25 | ASSERT_STREQ(metric->Name(), "pre");
|
30 |
| - EXPECT_NEAR(GetMetricEval(metric, {0, 1}, {0, 1}, {}, {}, data_split_mode), 0, 1e-7); |
31 |
| - EXPECT_NEAR(GetMetricEval(metric, |
32 |
| - {0.1f, 0.9f, 0.1f, 0.9f}, |
33 |
| - { 0, 0, 1, 1}, {}, {}, data_split_mode), |
34 |
| - 0, 1e-7); |
| 26 | + EXPECT_NEAR(GetMetricEval(metric, {0, 1}, {0, 1}, {}, {}, data_split_mode), 0.5, 1e-7); |
| 27 | + EXPECT_NEAR( |
| 28 | + GetMetricEval(metric, {0.1f, 0.9f, 0.1f, 0.9f}, {0, 0, 1, 1}, {}, {}, data_split_mode), 0.5, |
| 29 | + 1e-7); |
35 | 30 |
|
36 | 31 | delete metric;
|
37 | 32 | metric = xgboost::Metric::Create("pre@2", &ctx);
|
38 | 33 | ASSERT_STREQ(metric->Name(), "pre@2");
|
39 | 34 | EXPECT_NEAR(GetMetricEval(metric, {0, 1}, {0, 1}, {}, {}, data_split_mode), 0.5f, 1e-7);
|
40 |
| - EXPECT_NEAR(GetMetricEval(metric, |
41 |
| - {0.1f, 0.9f, 0.1f, 0.9f}, |
42 |
| - { 0, 0, 1, 1}, {}, {}, data_split_mode), |
43 |
| - 0.5f, 0.001f); |
| 35 | + EXPECT_NEAR( |
| 36 | + GetMetricEval(metric, {0.1f, 0.9f, 0.1f, 0.9f}, {0, 0, 1, 1}, {}, {}, data_split_mode), 0.5f, |
| 37 | + 0.001f); |
44 | 38 |
|
45 | 39 | EXPECT_ANY_THROW(GetMetricEval(metric, {0, 1}, {}, {}, {}, data_split_mode));
|
46 | 40 |
|
@@ -187,5 +181,4 @@ inline void VerifyNDCGExpGain(DataSplitMode data_split_mode = DataSplitMode::kRo
|
187 | 181 | ndcg = metric->Evaluate(predt, p_fmat);
|
188 | 182 | ASSERT_NEAR(ndcg, 1.0, kRtEps);
|
189 | 183 | }
|
190 |
| -} // namespace metric |
191 |
| -} // namespace xgboost |
| 184 | +} // namespace xgboost::metric |
0 commit comments