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