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.h
+++ b/tests/cpp/objective/test_lambdarank_obj.h
@@ -18,6 +18,8 @@
 #include "../helpers.h"                             // for EmptyDMatrix

 namespace xgboost::obj {
+void TestMAPStat(Context const* ctx);
+
 inline void TestNDCGJsonIO(Context const* ctx) {
  std::unique_ptr<xgboost::ObjFunction> obj{ObjFunction::Create("rank:ndcg", ctx)};

@@ -37,6 +39,8 @@ void TestNDCGGPair(Context const* ctx);

 void TestUnbiasedNDCG(Context const* ctx);

+void TestMAPGPair(Context const* ctx);
+
 /**
 * \brief Initialize test data for make pair tests.
 */