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

bug fix in pairwise rank

Browse Source
This commit is contained in:
tqchen
2014-05-15 15:37:58 -07:00
parent 9ea9a7a01e
commit 8f56671901
3 changed files with 45 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
demo/rank/mq2008.conf
View File

@@ -2,7 +2,7 @@
# choose the tree booster, 0: tree, 1: linear
booster_type = 0
#objective="rank:pairwise"
objective="rank:pairwise"
#objective="rank:softmax"
#objective="lambdarank:map"
#objective="lambdarank:ndcg"
@@ -15,8 +15,9 @@ bst:gamma = 1.0
# minimum sum of instance weight(hessian) needed in a child
bst:min_child_weight = 1
# maximum depth of a tree
bst:max_depth = 3
eval_metric='ndcg'
bst:max_depth = 6
eval_metric = "ndcg"
eval_metric = "map"
# Task parameters
# the number of round to do boosting
num_round = 4