7bccc1ea2c
- A new DMatrix type. - Extract common code into a new QDM base class. Not yet working: - Not exposed to the interface yet, will wait for the GPU implementation. - ~No meta info yet, still working on the source.~ - Exporting data to CSR is not supported yet.
39 lines
1.1 KiB
C++
39 lines
1.1 KiB
C++
/**
|
|
* Copyright 2023, XGBoost Contributors
|
|
*/
|
|
#ifndef XGBOOST_DATA_BATCH_UTILS_H_
|
|
#define XGBOOST_DATA_BATCH_UTILS_H_
|
|
|
|
#include "xgboost/data.h" // for BatchParam
|
|
|
|
namespace xgboost::data::detail {
|
|
// At least one batch parameter is initialized.
|
|
inline void CheckEmpty(BatchParam const& l, BatchParam const& r) {
|
|
if (!l.Initialized()) {
|
|
CHECK(r.Initialized()) << "Batch parameter is not initialized.";
|
|
}
|
|
}
|
|
|
|
/**
|
|
* \brief Should we regenerate the gradient index?
|
|
*
|
|
* \param old Parameter stored in DMatrix.
|
|
* \param p New parameter passed in by caller.
|
|
*/
|
|
inline bool RegenGHist(BatchParam old, BatchParam p) {
|
|
// Parameter is renewed or caller requests a regen
|
|
if (!p.Initialized()) {
|
|
// Empty parameter is passed in, don't regenerate so that we can use gindex in
|
|
// predictor, which doesn't have any training parameter.
|
|
return false;
|
|
}
|
|
return p.regen || old.ParamNotEqual(p);
|
|
}
|
|
|
|
/**
|
|
* @brief Validate the batch parameter from the caller
|
|
*/
|
|
void CheckParam(BatchParam const& init, BatchParam const& param);
|
|
} // namespace xgboost::data::detail
|
|
#endif // XGBOOST_DATA_BATCH_UTILS_H_
|