xgboost
Classes | Namespaces | Macros
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 <mutex>
Include dependency graph for predictor.h:
This graph shows which files directly or indirectly include this file:

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
 
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
 namespace of xgboost
 
 xgboost::gbm
 

Macros

#define XGBOOST_REGISTER_PREDICTOR(UniqueId, Name)
 

Detailed Description

Interface of predictor, performs predictions for a gradient booster.

Copyright 2017-2020 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)