Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

small change

Browse Source
This commit is contained in:
yepyao 2014-05-16 21:20:41 +08:00
parent dd71c0e070
commit a3bd5000ba
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
demo/rank/mq2008.conf
View File

@ -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

4
regrank/xgboost_regrank_obj.h
View File

@ -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;
}