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
728875b4
Commit
728875b4
authored
Oct 17, 2016
by
Guolin Ke
Committed by
GitHub
Oct 17, 2016
Browse files
fix check condition for min_sum_hessian_in_leaf and min_data_in_leaf
parent
3c6804cf
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
2 deletions
+1
-2
src/io/config.cpp
src/io/config.cpp
+1
-2
No files found.
src/io/config.cpp
View file @
728875b4
...
...
@@ -210,9 +210,8 @@ void MetricConfig::Set(const std::unordered_map<std::string, std::string>& param
void
TreeConfig
::
Set
(
const
std
::
unordered_map
<
std
::
string
,
std
::
string
>&
params
)
{
GetInt
(
params
,
"min_data_in_leaf"
,
&
min_data_in_leaf
);
CHECK
(
min_data_in_leaf
>
0
);
GetDouble
(
params
,
"min_sum_hessian_in_leaf"
,
&
min_sum_hessian_in_leaf
);
CHECK
(
min_sum_hessian_in_leaf
>
=
0.
0
);
CHECK
(
min_sum_hessian_in_leaf
>
1.0
f
||
min_data_in_leaf
>
0
);
GetInt
(
params
,
"num_leaves"
,
&
num_leaves
);
CHECK
(
num_leaves
>
0
);
GetInt
(
params
,
"feature_fraction_seed"
,
&
feature_fraction_seed
);
...
...
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