xgboost
Classes | Namespaces | Macros
gbm.h File Reference

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

#include <dmlc/registry.h>
#include <vector>
#include <utility>
#include <string>
#include <functional>
#include <memory>
#include "./base.h"
#include "./data.h"
#include "./objective.h"
#include "./feature_map.h"
#include "../../src/common/host_device_vector.h"
Include dependency graph for gbm.h:
This graph shows which files directly or indirectly include this file:

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