xgboost

Classes | Namespaces | Macros
tree_updater.h File Reference

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

`#include <dmlc/registry.h>`
`#include <xgboost/base.h>`
`#include <xgboost/data.h>`
`#include <xgboost/tree_model.h>`
`#include <xgboost/generic_parameters.h>`
`#include <xgboost/host_device_vector.h>`
`#include <xgboost/model.h>`
`#include <functional>`
`#include <vector>`
`#include <utility>`
`#include <string>`
Include dependency graph for tree_updater.h:

Go to the source code of this file.

## Classes
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
| xgboost
| namespace of xgboost

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

## Detailed Description

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

Copyright 2014-2019 by Contributors

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>();
});
XGBOOST_REGISTER_TREE_UPDATER
#define XGBOOST_REGISTER_TREE_UPDATER(UniqueId, Name)
Macro to register tree updater.
Definition: tree_updater.h:111