gbm.h 文件参考

梯度提升器的接口，通过梯度统计信息学习。 更多...

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

gbm.h 的包含依赖图

转到该文件的源代码。

类
class	xgboost::GradientBooster
	梯度提升模型的接口。 更多...
struct	xgboost::GradientBoosterReg
	树更新器的注册条目。 更多...

命名空间
	xgboost
	用于多目标树的核心数据结构。

宏
#define	XGBOOST_REGISTER_GBM(UniqueId, Name)
	用于注册梯度提升器的宏。 更多...

详细描述

梯度提升器的接口，通过梯度统计信息学习。

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

作者

Tianqi Chen

宏定义文档

XGBOOST_REGISTER_GBM

#define XGBOOST_REGISTER_GBM	(		UniqueId,
			Name
	)

值

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

用于注册梯度提升器的宏。

// example of registering a objective ndcg@k
XGBOOST_REGISTER_GBM(GBTree, "gbtree")
.describe("提升树集成。")
.set_body([]() {
    return new GradientBooster<TStats>();
  });