xgboost
|
Interface of predictor, performs predictions for a gradient booster. More...
#include <xgboost/base.h>
#include <xgboost/data.h>
#include <functional>
#include <memory>
#include <string>
#include <unordered_map>
#include <utility>
#include <vector>
#include "../../src/gbm/gbtree_model.h"
#include "../../src/common/host_device_vector.h"
Go to the source code of this file.
Classes | |
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::Predictor::PredictionCacheEntry |
Contains pointer to input matrix and associated cached predictions. More... | |
struct | xgboost::PredictorReg |
Registry entry for predictor. More... | |
Namespaces | |
xgboost | |
namespace of xgboost | |
Macros | |
#define | XGBOOST_REGISTER_PREDICTOR(UniqueId, Name) |
Interface of predictor, performs predictions for a gradient booster.
Copyright by Contributors
#define XGBOOST_REGISTER_PREDICTOR | ( | UniqueId, | |
Name | |||
) |