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>();
});
XGBOOST_REGISTER_GBM
#define XGBOOST_REGISTER_GBM(UniqueId, Name)
Macro to register gradient booster.
Definition: gbm.h:219