predictor.h File Reference

Interface of predictor, performs predictions for a gradient booster. More...

#include <dmlc/registry.h>
#include <xgboost/base.h>
#include <xgboost/cache.h>
#include <xgboost/context.h>
#include <xgboost/data.h>
#include <xgboost/host_device_vector.h>
#include <functional>
#include <memory>
#include <string>
#include <vector>

Include dependency graph for predictor.h:

Go to the source code of this file.

Classes
struct	xgboost::PredictionCacheEntry
	Contains pointer to input matrix and associated cached predictions. More...
class	xgboost::PredictionContainer
	A container for managed prediction caches. More...
class	xgboost::Predictor
	Performs prediction on individual training instances or batches of instances for GBTree. Prediction functions all take a GBTreeModel and a DMatrix as input and output a vector of predictions. The predictor does not modify any state of the model itself. More...
struct	xgboost::PredictorReg
	Registry entry for predictor. More...

Namespaces
	xgboost
	Learner interface that integrates objective, gbm and evaluation together. This is the user facing XGBoost training module.
	xgboost::gbm

Macros
#define	XGBOOST_REGISTER_PREDICTOR(UniqueId, Name)

Detailed Description

Interface of predictor, performs predictions for a gradient booster.

Copyright 2017-2025, XGBoost Contributors

Macro Definition Documentation

XGBOOST_REGISTER_PREDICTOR

#define XGBOOST_REGISTER_PREDICTOR	(		UniqueId,
			Name
	)

Value:

  static DMLC_ATTRIBUTE_UNUSED ::xgboost::PredictorReg& __make_##PredictorReg##_##UniqueId##__ = \
      ::dmlc::Registry<::xgboost::PredictorReg>::Get()->__REGISTER__(Name)