tree_updater.h 文件参考

用于树学习的通用原语，根据信息更新树的集合。更多...

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

tree_updater.h 的包含依赖图

转到此文件的源代码。

类
class	xgboost::TreeUpdater
	树更新模块的接口，用于执行树的更新。更多...
struct	xgboost::TreeUpdaterReg
	树更新器的注册表项。更多...

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

宏
#define	XGBOOST_REGISTER_TREE_UPDATER(UniqueId, Name)
	注册树更新器的宏。更多...

详细描述

用于树学习的通用原语，根据信息更新树的集合。

版权所有 2014-2023 XGBoost 贡献者

作者

Tianqi Chen

宏定义文档

XGBOOST_REGISTER_TREE_UPDATER

#define XGBOOST_REGISTER_TREE_UPDATER	(		UniqueId,
			Name
	)

值

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

注册树更新器的宏。

// 注册目标函数 ndcg@k 的示例
XGBOOST_REGISTER_TREE_UPDATER(ColMaker, "colmaker")
.describe("基于列的树构建器。")
.set_body([]() {
    return new ColMaker<TStats>();
  });