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

chg begin end type

Browse Source
This commit is contained in:
tqchen
2014-11-10 17:24:44 -08:00
parent 9855a90142
commit b426eef527
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/utils/quantile.h
View File

@@ -165,15 +165,16 @@ struct WQSummary {
if (src.size <= maxsize) {
this->CopyFrom(src); return;
}
const RType max_rank = src.MaxRank();
const RType begin = src.data[0].rmax;
const RType range = src.data[src.size - 1].rmin - src.data[0].rmax;
const size_t n = maxsize - 1;
data[0] = src.data[0];
this->size = 1;
// lastidx is used to avoid duplicated records
size_t i = 0, lastidx = 0;
size_t i = 1, lastidx = 0;
for (RType k = 1; k < n; ++k) {
RType dx2 = (2 * k * max_rank) / n;
// find first i such that d < (rmax[i+1] + rmin[i+1]) / 2
RType dx2 = 2 * ((k * range) / n + begin);
// find first i such that d < (rmax[i+1] + rmin[i+1]) / 2
while (i < src.size - 1 &&
dx2 >= src.data[i + 1].rmax + src.data[i + 1].rmin) ++i;
if (i == src.size - 1) break;