Inheritance diagram for xgboost::predictor::PredictorOneAPI:

Public Member Functions
	PredictorOneAPI (Context const *generic_param)

void	PredictBatch (DMatrix dmat, PredictionCacheEntry predts, const gbm::GBTreeModel &model, int tree_begin, uint32_t const ntree_limit=0) override

void	InplacePredict (std::any const &x, const gbm::GBTreeModel &model, float missing, PredictionCacheEntry *out_preds, uint32_t tree_begin, unsigned tree_end) const override

void	PredictInstance (const SparsePage::Inst &inst, std::vector< bst_float > *out_preds, const gbm::GBTreeModel &model, unsigned ntree_limit) override

void	PredictLeaf (DMatrix p_fmat, std::vector< bst_float > out_preds, const gbm::GBTreeModel &model, unsigned ntree_limit) override

void	PredictContribution (DMatrix p_fmat, std::vector< bst_float > out_contribs, const gbm::GBTreeModel &model, uint32_t ntree_limit, std::vector< bst_float > *tree_weights, bool approximate, int condition, unsigned condition_feature) override

void	PredictInteractionContributions (DMatrix p_fmat, std::vector< bst_float > out_contribs, const gbm::GBTreeModel &model, unsigned ntree_limit, std::vector< bst_float > *tree_weights, bool approximate) override

Public Member Functions inherited from xgboost::Predictor
	Predictor (Context const *ctx)

virtual void	Configure (Args const &)
	Configure and register input matrices in prediction cache.

void	InitOutPredictions (const MetaInfo &info, HostDeviceVector< bst_float > *out_predt, const gbm::GBTreeModel &model) const
	Initialize output prediction.

virtual void	PredictBatch (DMatrix dmat, PredictionCacheEntry out_preds, const gbm::GBTreeModel &model, uint32_t tree_begin, uint32_t tree_end=0) const =0
	Generate batch predictions for a given feature matrix.

virtual bool	InplacePredict (std::shared_ptr< DMatrix > p_fmat, const gbm::GBTreeModel &model, float missing, PredictionCacheEntry *out_preds, uint32_t tree_begin=0, uint32_t tree_end=0) const =0
	Inplace prediction.

virtual void	PredictInstance (const SparsePage::Inst &inst, std::vector< bst_float > *out_preds, const gbm::GBTreeModel &model, unsigned tree_end=0, bool is_column_split=false) const =0
	online prediction function, predict score for one instance at a time NOTE: use the batch prediction interface if possible, batch prediction is usually more efficient than online prediction This function is NOT threadsafe, make sure you only call from one thread.

virtual void	PredictLeaf (DMatrix dmat, HostDeviceVector< bst_float > out_preds, const gbm::GBTreeModel &model, unsigned tree_end=0) const =0
	predict the leaf index of each tree, the output will be nsample * ntree vector this is only valid in gbtree predictor.

virtual void	PredictContribution (DMatrix dmat, HostDeviceVector< bst_float > out_contribs, const gbm::GBTreeModel &model, unsigned tree_end=0, std::vector< bst_float > const *tree_weights=nullptr, bool approximate=false, int condition=0, unsigned condition_feature=0) const =0
	feature contributions to individual predictions; the output will be a vector of length (nfeats + 1) * num_output_group * nsample, arranged in that order.

virtual void	PredictInteractionContributions (DMatrix dmat, HostDeviceVector< bst_float > out_contribs, const gbm::GBTreeModel &model, unsigned tree_end=0, std::vector< bst_float > const *tree_weights=nullptr, bool approximate=false) const =0

Protected Member Functions
void	InitOutPredictions (const MetaInfo &info, HostDeviceVector< bst_float > *out_preds, const gbm::GBTreeModel &model) const

void	DevicePredictInternal (DeviceMatrixOneAPI dmat, HostDeviceVector< float > out_preds, const gbm::GBTreeModel &model, size_t tree_begin, size_t tree_end)

Additional Inherited Members
Static Public Member Functions inherited from xgboost::Predictor
static Predictor *	Create (std::string const &name, Context const *ctx)
	Creates a new Predictor*.

Protected Attributes inherited from xgboost::Predictor
Context const *	ctx_

The documentation for this class was generated from the following file:

External/xgboost/plugin/updater_oneapi/predictor_oneapi.cc

Public Member Functions

Protected Member Functions

Additional Inherited Members