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
fb8d2eb7
Commit
fb8d2eb7
authored
Jan 11, 2018
by
Zibi789
Committed by
Guolin Ke
Jan 11, 2018
Browse files
It fixes the bug in ResetGoss (issue #1192) (#1193)
parent
195d9531
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
0 deletions
+2
-0
src/boosting/goss.hpp
src/boosting/goss.hpp
+2
-0
No files found.
src/boosting/goss.hpp
View file @
fb8d2eb7
...
@@ -14,6 +14,7 @@
...
@@ -14,6 +14,7 @@
#include <string>
#include <string>
#include <fstream>
#include <fstream>
#include <chrono>
#include <chrono>
#include <algorithm>
namespace
LightGBM
{
namespace
LightGBM
{
...
@@ -75,6 +76,7 @@ public:
...
@@ -75,6 +76,7 @@ public:
is_use_subset_
=
false
;
is_use_subset_
=
false
;
if
(
gbdt_config_
->
top_rate
+
gbdt_config_
->
other_rate
<=
0.5
)
{
if
(
gbdt_config_
->
top_rate
+
gbdt_config_
->
other_rate
<=
0.5
)
{
auto
bag_data_cnt
=
static_cast
<
data_size_t
>
((
gbdt_config_
->
top_rate
+
gbdt_config_
->
other_rate
)
*
num_data_
);
auto
bag_data_cnt
=
static_cast
<
data_size_t
>
((
gbdt_config_
->
top_rate
+
gbdt_config_
->
other_rate
)
*
num_data_
);
bag_data_cnt
=
std
::
max
(
1
,
bag_data_cnt
);
tmp_subset_
.
reset
(
new
Dataset
(
bag_data_cnt
));
tmp_subset_
.
reset
(
new
Dataset
(
bag_data_cnt
));
tmp_subset_
->
CopyFeatureMapperFrom
(
train_data_
);
tmp_subset_
->
CopyFeatureMapperFrom
(
train_data_
);
is_use_subset_
=
true
;
is_use_subset_
=
true
;
...
...
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