Medial Code Documentation
|
Copyright 2017-2023 by XGBoost Contributors. More...
Data Structures | |
class | ColumnSplitHelper |
A helper class for prediction when the DMatrix is split by column. More... | |
class | CPUPredictor |
struct | DeviceMatrixOneAPI |
class | DeviceModelOneAPI |
struct | DeviceNodeOneAPI |
struct | EntryOneAPI |
Element from a sparse vector. More... | |
class | PredictorOneAPI |
Functions | |
DMLC_REGISTRY_FILE_TAG (predictor_oneapi) | |
float | GetFvalue (int ridx, int fidx, EntryOneAPI *data, size_t *row_ptr, bool &is_missing) |
float | GetLeafWeight (int ridx, const DeviceNodeOneAPI *tree, EntryOneAPI *data, size_t *row_ptr) |
describe ("Make predictions using DPC++.") .set_body([](Context const *generic_param) | |
DMLC_REGISTRY_FILE_TAG (cpu_predictor) | |
describe ("Make predictions using CPU.") .set_body([](Context const *ctx) | |
template<bool has_categorical> | |
XGBOOST_DEVICE bool | GetDecision (RegTree::Node const &node, bst_node_t nid, float fvalue, RegTree::CategoricalSplitMatrix const &cats) |
Whether it should traverse to the left branch of a tree. | |
template<bool has_missing, bool has_categorical> | |
XGBOOST_DEVICE bst_node_t | GetNextNode (const RegTree::Node &node, const bst_node_t nid, float fvalue, bool is_missing, RegTree::CategoricalSplitMatrix const &cats) |
template<bool has_missing, bool has_categorical> | |
XGBOOST_DEVICE bst_node_t | GetNextNodeMulti (MultiTargetTree const &tree, bst_node_t const nidx, float fvalue, bool is_missing, RegTree::CategoricalSplitMatrix const &cats) |
DMLC_REGISTRY_LINK_TAG (cpu_predictor) | |
Copyright 2017-2023 by XGBoost Contributors.
Copyright 2021-2023 by XGBoost Contributors.