diff --git a/demo/rank/mq2008.conf b/demo/rank/mq2008.conf index 08b9d9679..0853fcb4e 100644 --- a/demo/rank/mq2008.conf +++ b/demo/rank/mq2008.conf @@ -4,8 +4,8 @@ booster_type = 0 objective="rank:pairwise" #objective="rank:softmax" -#objective="lambdarank:map" -#objective="lambdarank:ndcg" +#objective="rank:map" +#objective="rank:ndcg" # Tree Booster Parameters # step size shrinkage diff --git a/regrank/xgboost_regrank_obj.h b/regrank/xgboost_regrank_obj.h index f2fee0653..5a7bad9dd 100644 --- a/regrank/xgboost_regrank_obj.h +++ b/regrank/xgboost_regrank_obj.h @@ -113,8 +113,8 @@ namespace xgboost{ if( !strcmp("rank:pairwise", name ) ) return new PairwiseRankObj(); if( !strcmp("rank:softmax", name ) ) return new SoftmaxRankObj(); if( !strcmp("softmax", name ) ) return new SoftmaxMultiClassObj(); - // if (!strcmp("lambdarank:map", name)) return new LambdaRankObj_MAP(); - // if (!strcmp("lambdarank:ndcg", name)) return new LambdaRankObj_NDCG(); + if (!strcmp("rank:map", name)) return new LambdaRankObj_MAP(); + if (!strcmp("rank:ndcg", name)) return new LambdaRankObj_NDCG(); utils::Error("unknown objective function type"); return NULL; }