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

fix Dart::NormalizeTrees (#1265)

Browse Source
This commit is contained in:
Yoshinori Nakano
2016-06-10 07:28:24 +09:00
committed by Tianqi Chen
parent 949d1e3027
commit 7cfeb5f012
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/gbm/gbtree.cc
View File

@@ -703,7 +703,7 @@ class Dart : public GBTree {
weight_drop[i] *= factor;
}
for (size_t i = 0; i < size_new_trees; ++i) {
weight_drop.push_back(lr * factor);
weight_drop.push_back(factor);
}
} else {
// normalize_type 0
@@ -712,7 +712,7 @@ class Dart : public GBTree {
weight_drop[i] *= factor;
}
for (size_t i = 0; i < size_new_trees; ++i) {
weight_drop.push_back(1.0 * lr / (num_drop + lr));
weight_drop.push_back(1.0 / (num_drop + lr));
}
}
}