predictor.h File Reference

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

#include <xgboost/base.h>
#include <xgboost/data.h>
#include <xgboost/generic_parameters.h>
#include <xgboost/host_device_vector.h>
#include <functional>
#include <memory>
#include <string>
#include <unordered_map>
#include <utility>
#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::Predictor
	Performs prediction on individual training instances or batches of instances for GBTree. The predictor also manages a prediction cache associated with input matrices. If possible, it will use previously calculated predictions instead of calculating new predictions. 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
	namespace of xgboost
	xgboost::gbm

Macros
#define	XGBOOST_REGISTER_PREDICTOR(UniqueId, Name)

Detailed Description

Interface of predictor, performs predictions for a gradient booster.

Copyright by 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)