tree_updater.h File Reference

Copyright 2014-2023 by XGBoost Contributors. More...

#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>

Go to the source code of this file.

Data Structures
class	xgboost::TreeUpdater
	interface of tree update module, that performs update of a tree. More...
struct	xgboost::TreeUpdaterReg
	Registry entry for tree updater. More...

Namespaces
namespace	xgboost
	namespace of xgboost
namespace	xgboost::tree
	Copyright 2021-2023 by XGBoost Contributors.

Macros
#define	XGBOOST_REGISTER_TREE_UPDATER(UniqueId, Name)
	Macro to register tree updater.

Detailed Description

Copyright 2014-2023 by XGBoost Contributors.

General primitive for tree learning, Updating a collection of trees given the information.

Author

Tianqi Chen

Macro Definition Documentation

XGBOOST_REGISTER_TREE_UPDATER

#define XGBOOST_REGISTER_TREE_UPDATER	(		UniqueId,
			Name
	)

Value:

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

Macro to register tree updater.

// example of registering a objective ndcg@k
XGBOOST_REGISTER_TREE_UPDATER(ColMaker, "colmaker")
.describe("Column based tree maker.")
.set_body([]() {
    return new ColMaker<TStats>();
  });