Medial Code Documentation
Loading...
Searching...
No Matches
batch_utils.h
1
4#ifndef XGBOOST_DATA_BATCH_UTILS_H_
5#define XGBOOST_DATA_BATCH_UTILS_H_
6
7#include "xgboost/data.h" // for BatchParam
8
9namespace xgboost::data::detail {
10// At least one batch parameter is initialized.
11inline void CheckEmpty(BatchParam const& l, BatchParam const& r) {
12 if (!l.Initialized()) {
13 CHECK(r.Initialized()) << "Batch parameter is not initialized.";
14 }
15}
16
23inline bool RegenGHist(BatchParam old, BatchParam p) {
24 // Parameter is renewed or caller requests a regen
25 if (!p.Initialized()) {
26 // Empty parameter is passed in, don't regenerate so that we can use gindex in
27 // predictor, which doesn't have any training parameter.
28 return false;
29 }
30 return p.regen || old.ParamNotEqual(p);
31}
32} // namespace xgboost::data::detail
33#endif // XGBOOST_DATA_BATCH_UTILS_H_
Copyright 2015-2023 by XGBoost Contributors.
Copyright 2023, XGBoost Contributors.
Definition adapter.h:106
bool RegenGHist(BatchParam old, BatchParam p)
Should we regenerate the gradient index?
Definition batch_utils.h:23
Parameters for constructing histogram index batches.
Definition data.h:244