Rework MAP and Pairwise for LTR. (#9075)

2023-04-28 02:39:12 +08:00
parent 0e470ef606
commit e206b899ef
19 changed files with 612 additions and 1135 deletions
--- a/tests/cpp/objective/test_lambdarank_obj.cu
+++ b/tests/cpp/objective/test_lambdarank_obj.cu
@@ -18,6 +18,12 @@ TEST(LambdaRank, GPUNDCGJsonIO) {
  TestNDCGJsonIO(&ctx);
 }

+TEST(LambdaRank, GPUMAPStat) {
+  Context ctx;
+  ctx.gpu_id = 0;
+  TestMAPStat(&ctx);
+}
+
 TEST(LambdaRank, GPUNDCGGPair) {
  Context ctx;
  ctx.gpu_id = 0;
@@ -153,4 +159,10 @@ TEST(LambdaRank, RankItemCountOnRight) {
  RankItemCountImpl(sorted_items, wrapper, 1, static_cast<uint32_t>(1));
  RankItemCountImpl(sorted_items, wrapper, 0, static_cast<uint32_t>(0));
 }
+
+TEST(LambdaRank, GPUMAPGPair) {
+  Context ctx;
+  ctx.gpu_id = 0;
+  TestMAPGPair(&ctx);
+}
 }  // namespace xgboost::obj