Define pair generation strategies for LTR. (#8984)

2023-03-30 12:00:35 +08:00
parent d385cc64e2
commit d062a9e009
11 changed files with 770 additions and 48 deletions
--- a/src/common/ranking_utils.h
+++ b/src/common/ranking_utils.h
@@ -123,7 +123,7 @@ struct LambdaRankParam : public XGBoostParameter<LambdaRankParam> {

  DMLC_DECLARE_PARAMETER(LambdaRankParam) {
    DMLC_DECLARE_FIELD(lambdarank_pair_method)
-        .set_default(PairMethod::kMean)
+        .set_default(PairMethod::kTopK)
        .add_enum("mean", PairMethod::kMean)
        .add_enum("topk", PairMethod::kTopK)
        .describe("Method for constructing pairs.");