|
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