Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix dart bug (#1882)

Browse Source
This commit is contained in:
wxchan 2016-12-20 01:01:28 +08:00 committed by Tianqi Chen
parent fa97259d66
commit cee4aafb93
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/gbm/gbtree.cc
View File

@ -680,7 +680,7 @@ class Dart : public GBTree {
// normalize_type 1
float factor = 1.0 / (1.0 + lr);
for (size_t i = 0; i < idx_drop.size(); ++i) {
weight_drop[i] *= factor;
weight_drop[idx_drop[i]] *= factor;
}
for (size_t i = 0; i < size_new_trees; ++i) {
weight_drop.push_back(factor);
@ -689,7 +689,7 @@ class Dart : public GBTree {
// normalize_type 0
float factor = 1.0 * num_drop / (num_drop + lr);
for (size_t i = 0; i < idx_drop.size(); ++i) {
weight_drop[i] *= factor;
weight_drop[idx_drop[i]] *= factor;
}
for (size_t i = 0; i < size_new_trees; ++i) {
weight_drop.push_back(1.0 / (num_drop + lr));