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
c4536e22
Unverified
Commit
c4536e22
authored
Feb 15, 2020
by
Guolin Ke
Committed by
GitHub
Feb 15, 2020
Browse files
fix (#2751)
Co-authored-by:
Nikita Titov
<
nekit94-08@mail.ru
>
parent
923226b1
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
1 deletion
+4
-1
src/treelearner/cost_effective_gradient_boosting.hpp
src/treelearner/cost_effective_gradient_boosting.hpp
+4
-1
No files found.
src/treelearner/cost_effective_gradient_boosting.hpp
View file @
c4536e22
...
...
@@ -70,8 +70,11 @@ class CostEfficientGradientBoosting {
if
(
i
==
best_leaf
)
continue
;
auto
split
=
&
splits_per_leaf_
[
static_cast
<
size_t
>
(
i
)
*
train_data
->
num_features
()
+
inner_feature_index
];
split
->
gain
+=
config
->
cegb_tradeoff
*
config
->
cegb_penalty_feature_coupled
[
best_split_info
->
feature
];
if
(
*
split
>
ref_best_split_per_leaf
[
i
])
// Avoid to update the leaf that cannot split
if
(
ref_best_split_per_leaf
[
i
].
gain
>
kMinScore
&&
*
split
>
ref_best_split_per_leaf
[
i
])
{
ref_best_split_per_leaf
[
i
]
=
*
split
;
}
}
}
if
(
!
config
->
cegb_penalty_feature_lazy
.
empty
())
{
...
...
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