gbm.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/model.h>
#include <vector>
#include <utility>
#include <string>
#include <functional>
#include <unordered_map>
#include <memory>

Go to the source code of this file.

Data Structures
class	xgboost::GradientBooster
	interface of gradient boosting model. More...
struct	xgboost::GradientBoosterReg
	Registry entry for tree updater. More...

Namespaces
namespace	xgboost
	namespace of xgboost

Macros
#define	XGBOOST_REGISTER_GBM(UniqueId, Name)
	Macro to register gradient booster.

Detailed Description

Copyright 2014-2023 by XGBoost Contributors.

Interface of gradient booster, that learns through gradient statistics.

Author

Tianqi Chen

Macro Definition Documentation

XGBOOST_REGISTER_GBM

#define XGBOOST_REGISTER_GBM	(		UniqueId,
			Name
	)

Value:

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

Macro to register gradient booster.

// example of registering a objective ndcg@k
XGBOOST_REGISTER_GBM(GBTree, "gbtree")
.describe("Boosting tree ensembles.")
.set_body([]() {
    return new GradientBooster<TStats>();
  });