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

Merge extract cuts into QuantileContainer. (#6125)

Browse Source 
* Use pruning for initial summary construction.
This commit is contained in:
Jiaming Yuan
2020-09-18 16:36:39 +08:00
committed by GitHub
parent cc82ca167a
commit e319b63f9e
6 changed files with 171 additions and 210 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/common/quantile.cuh
View File

@@ -96,13 +96,21 @@ class SketchContainer {
* addition inside `RMinNext` and subtraction in `RMaxPrev`. */
void FixError();
/* \brief Push a CSC structured cut matrix. */
void Push(common::Span<OffsetT const> cuts_ptr,
dh::caching_device_vector<SketchEntry>* entries);
/* \brief Push sorted entries.
*
* \param entries Sorted entries.
* \param columns_ptr CSC pointer for entries.
* \param cuts_ptr CSC pointer for cuts.
* \param total_cuts Total number of cuts, equal to the back of cuts_ptr.
* \param weights (optional) data weights.
*/
void Push(Span<Entry const> entries, Span<size_t> columns_ptr,
common::Span<OffsetT const> cuts_ptr, size_t total_cuts,
Span<float> weights = {});
/* \brief Prune the quantile structure.
*
* \param to The maximum size of pruned quantile. If the size of quantile structure is
* already less than `to`, then no operation is performed.
* \param to The maximum size of pruned quantile. If the size of quantile
* structure is already less than `to`, then no operation is performed.
*/
void Prune(size_t to);
/* \brief Merge another set of sketch.
@@ -135,8 +143,8 @@ struct SketchUnique {
return a.value - b.value == 0;
}
};
} // anonymous detail
} // namespace detail
} // namespace common
} // namespace xgboost
#endif // XGBOOST_COMMON_QUANTILE_CUH_
#endif // XGBOOST_COMMON_QUANTILE_CUH_