34 lines
937 B
Plaintext
34 lines
937 B
Plaintext
# General Parameters, see comment for each definition
|
|
# choose the tree booster, 0: tree, 1: linear
|
|
booster_type = 0
|
|
|
|
objective="rank:pairwise"
|
|
#objective="rank:softmax"
|
|
#objective="lambdarank:map"
|
|
#objective="lambdarank:ndcg"
|
|
|
|
# Tree Booster Parameters
|
|
# step size shrinkage
|
|
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 = 0.1
|
|
# maximum depth of a tree
|
|
bst:max_depth = 6
|
|
eval_metric = "ndcg"
|
|
eval_metric = "map"
|
|
# Task parameters
|
|
# the number of round to do boosting
|
|
num_round = 4
|
|
# 0 means do not save any model except the final round model
|
|
save_period = 0
|
|
# The path of training data
|
|
data = "mq2008.train"
|
|
# The path of validation data, used to monitor training process, here [test] sets name of the validation set
|
|
eval[test] = "mq2008.vali"
|
|
# The path of test data
|
|
test:data = "mq2008.test"
|
|
|
|
|