Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
tianlh
LightGBM-DCU
Commits
18d6a902
Commit
18d6a902
authored
Apr 13, 2017
by
Guolin Ke
Browse files
change defualt values for some parameters.
parent
3a8b5e5f
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
7 deletions
+7
-7
include/LightGBM/config.h
include/LightGBM/config.h
+7
-7
No files found.
include/LightGBM/config.h
View file @
18d6a902
...
...
@@ -108,9 +108,9 @@ public:
int
bin_construct_sample_cnt
=
200000
;
bool
is_predict_leaf_index
=
false
;
bool
is_predict_raw_score
=
false
;
int
min_data_in_leaf
=
10
0
;
int
min_data_in_leaf
=
2
0
;
int
min_data_in_bin
=
5
;
double
max_conflict_rate
=
0.
000
0
f
;
double
max_conflict_rate
=
0.0
f
;
bool
enable_bundle
=
true
;
bool
adjacent_bundle
=
false
;
bool
has_header
=
false
;
...
...
@@ -176,13 +176,13 @@ public:
/*! \brief Config for tree model */
struct
TreeConfig
:
public
ConfigBase
{
public:
int
min_data_in_leaf
=
10
0
;
double
min_sum_hessian_in_leaf
=
1
0.0
f
;
int
min_data_in_leaf
=
2
0
;
double
min_sum_hessian_in_leaf
=
1
e-3
f
;
double
lambda_l1
=
0.0
f
;
double
lambda_l2
=
0.0
f
;
double
min_gain_to_split
=
0.0
f
;
// should > 1
, only one leaf means not need to learning
int
num_leaves
=
1
27
;
// should > 1
int
num_leaves
=
3
1
;
int
feature_fraction_seed
=
2
;
double
feature_fraction
=
1.0
f
;
// max cache size(unit:MB) for historical histogram. < 0 means no limit
...
...
@@ -213,7 +213,7 @@ public:
double
sigmoid
=
1.0
f
;
int
output_freq
=
1
;
bool
is_provide_training_metric
=
false
;
int
num_iterations
=
10
;
int
num_iterations
=
10
0
;
double
learning_rate
=
0.1
f
;
double
bagging_fraction
=
1.0
f
;
int
bagging_seed
=
3
;
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment