xgboost
Classes | Namespaces | Macros
gbm.h File Reference

Interface of gradient booster, that learns through gradient statistics. More...

#include <dmlc/registry.h>
#include <dmlc/any.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>
Include dependency graph for gbm.h:

Go to the source code of this file.

Classes

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

Namespaces

 xgboost
 namespace of xgboost
 

Macros

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

Detailed Description

Interface of gradient booster, that learns through gradient statistics.

Copyright 2014-2020 by Contributors

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>();
});