"vscode:/vscode.git/clone" did not exist on "702fff227d8960e532dd9ad4abd8e4456084345a"
train_linear.conf 3.28 KB
Newer Older
1
2
3
4
5
6
7
8
9
# task type, support train and predict
task = train

# boosting type, support gbdt for now, alias: boosting, boost
boosting_type = gbdt

# application type, support following application
# regression , regression task
# binary , binary classification task
Andrew Ziem's avatar
Andrew Ziem committed
10
# lambdarank , LambdaRank task
11
12
13
14
15
# alias: application, app
objective = binary

linear_tree = true

Andrew Ziem's avatar
Andrew Ziem committed
16
# eval metrics, support multi metric, delimited by ',' , support following metrics
17
18
19
20
21
22
23
24
# l1 
# l2 , default metric for regression
# ndcg , default metric for lambdarank
# auc 
# binary_logloss , default metric for binary
# binary_error
metric = binary_logloss,auc

Andrew Ziem's avatar
Andrew Ziem committed
25
# frequency for metric output
26
27
28
29
30
31
32
33
34
metric_freq = 1

# true if need output metric for training data, alias: tranining_metric, train_metric
is_training_metric = true

# number of bins for feature bucket, 255 is a recommend setting, it can save memories, and also has good accuracy. 
max_bin = 255

# training data
Andrew Ziem's avatar
Andrew Ziem committed
35
# if existing weight file, should name to "binary.train.weight"
36
37
38
39
# alias: train_data, train
data = binary.train

# validation data, support multi validation data, separated by ','
Andrew Ziem's avatar
Andrew Ziem committed
40
# if existing weight file, should name to "binary.test.weight"
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
# alias: valid, test, test_data, 
valid_data = binary.test

# number of trees(iterations), alias: num_tree, num_iteration, num_iterations, num_round, num_rounds
num_trees = 100

# shrinkage rate , alias: shrinkage_rate
learning_rate = 0.1

# number of leaves for one tree, alias: num_leaf
num_leaves = 63

# type of tree learner, support following types:
# serial , single machine version
# feature , use feature parallel to train
# data , use data parallel to train
# voting , use voting based parallel to train
# alias: tree
tree_learner = serial

Andrew Ziem's avatar
Andrew Ziem committed
61
# number of threads for multi-threading. One thread will use each CPU. The default is set to CPU count. 
62
63
64
65
66
67
68
69
70
# num_threads = 8

# feature sub-sample, will random select 80% feature to train on each iteration 
# alias: sub_feature
feature_fraction = 0.8

# Support bagging (data sub-sample), will perform bagging every 5 iterations
bagging_freq = 5

Andrew Ziem's avatar
Andrew Ziem committed
71
# Bagging fraction, will random select 80% data on bagging
72
73
74
75
76
77
78
# alias: sub_row
bagging_fraction = 0.8

# minimal number data for one leaf, use this to deal with over-fit
# alias : min_data_per_leaf, min_data
min_data_in_leaf = 50

Andrew Ziem's avatar
Andrew Ziem committed
79
# minimal sum Hessians for one leaf, use this to deal with over-fit
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
min_sum_hessian_in_leaf = 5.0

# save memory and faster speed for sparse feature, alias: is_sparse
is_enable_sparse = true

# when data is bigger than memory size, set this to true. otherwise set false will have faster speed
# alias: two_round_loading, two_round
use_two_round_loading = false

# true if need to save data to binary file and application will auto load data from binary file next time
# alias: is_save_binary, save_binary
is_save_binary_file = false

# output model file
output_model = LightGBM_model.txt

# support continuous train from trained gbdt model
# input_model= trained_model.txt

# output prediction file for predict task
# output_result= prediction.txt


103
# number of machines in distributed training, alias: num_machine
104
105
num_machines = 1

106
# local listening port in distributed training, alias: local_port
107
108
local_listen_port = 12400

109
# machines list file for distributed training, alias: mlist
110
111
112
113
machine_list_file = mlist.txt

# force splits
# forced_splits = forced_splits.json