xgboost
| 命名空间 |
predictor.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>
predictor.h 的包含依赖图

转到此文件的源代码。

struct  xgboost::PredictionCacheEntry
 包含指向输入矩阵及相关缓存预测的指针。 更多...
 
class  xgboost::PredictionContainer
 托管预测缓存的容器。 更多...
 
class  xgboost::Predictor
 对 GBTree 的单个训练实例或批次实例执行预测。所有预测函数都接受 GBTreeModel 和一个 DMatrix 作为输入,并输出一个预测向量。预测器不会修改模型本身的任何状态。 更多...
 
struct  xgboost::PredictorReg
 预测器的注册表条目。 更多...
 

命名空间

 xgboost
 多目标树的核心数据结构。
 
 xgboost::gbm
 

#define XGBOOST_REGISTER_PREDICTOR(UniqueId, Name)
 

详细描述

预测器接口,用于执行梯度提升器的预测。

版权所有 2017-2025,XGBoost 贡献者

宏定义文档

◆ XGBOOST_REGISTER_PREDICTOR

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