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

chg rank demo

Browse Source
This commit is contained in:
tqchen 2014-05-16 18:38:40 -07:00
parent 4743cc98ec
commit bee87cfce7
2 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
demo/rank/mq2008.conf
View File

@ -3,9 +3,6 @@
booster_type = 0 booster_type = 0
objective="rank:pairwise" objective="rank:pairwise"
#objective="rank:softmax"
#objective="lambdarank:map"
#objective="lambdarank:ndcg"
# Tree Booster Parameters # Tree Booster Parameters
# step size shrinkage # step size shrinkage
@ -16,8 +13,7 @@ bst:gamma = 1.0
bst:min_child_weight = 0.1 bst:min_child_weight = 0.1
# maximum depth of a tree # maximum depth of a tree
bst:max_depth = 6 bst:max_depth = 6
eval_metric = "ndcg"
eval_metric = "map"
# Task parameters # Task parameters
# the number of round to do boosting # the number of round to do boosting
num_round = 4 num_round = 4

1
demo/rank/runexp.sh
View File

@ -8,4 +8,3 @@ python trans_data.py vali.txt mq2008.vali mq2008.vali.group
../../xgboost mq2008.conf task=pred model_in=0002.model ../../xgboost mq2008.conf task=pred model_in=0002.model
../../xgboost mq2008.conf task=dump model_in=0002.model name_dump=dump.raw.txt