This commit is contained in:
tqchen
2014-05-15 18:56:28 -07:00
parent 8f56671901
commit 698fa87bc3
3 changed files with 46 additions and 36 deletions

View File

@@ -2,9 +2,9 @@
# 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:map"
#objective="lambdarank:ndcg"
# Tree Booster Parameters
@@ -13,7 +13,7 @@ bst:eta = 0.1
# minimum loss reduction required to make a further partition
bst:gamma = 1.0
# minimum sum of instance weight(hessian) needed in a child
bst:min_child_weight = 1
bst:min_child_weight = 0.1
# maximum depth of a tree
bst:max_depth = 6
eval_metric = "ndcg"