From a3bd5000bab460ed9a97cb44d137c9e76913c7aa Mon Sep 17 00:00:00 2001 From: yepyao Date: Fri, 16 May 2014 21:20:41 +0800 Subject: [PATCH] small change --- demo/rank/mq2008.conf | 4 ++-- regrank/xgboost_regrank_obj.h | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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; }