tree_model.cc File Reference

Copyright 2015-2023 by Contributors. More...

#include <dmlc/json.h>
#include <dmlc/registry.h>
#include <xgboost/json.h>
#include <xgboost/tree_model.h>
#include <cmath>
#include <iomanip>
#include <limits>
#include <sstream>
#include <type_traits>
#include "../common/categorical.h"
#include "../common/common.h"
#include "../predictor/predict_fn.h"
#include "io_utils.h"
#include "param.h"
#include "xgboost/base.h"
#include "xgboost/data.h"
#include "xgboost/logging.h"

Data Structures
class	xgboost::TreeGenerator
	Base class for dump model implementation, modeling closely after code generator. More...
struct	xgboost::TreeGenReg
class	xgboost::TextGenerator
class	xgboost::JsonGenerator
struct	xgboost::GraphvizParam
class	xgboost::GraphvizGenerator

Namespaces
namespace	xgboost
	namespace of xgboost
namespace	xgboost::tree
	Copyright 2021-2023 by XGBoost Contributors.
namespace	dmlc
	namespace for dmlc

Macros
#define	XGBOOST_REGISTER_TREE_IO(UniqueId, Name)

Functions
	xgboost::DMLC_REGISTER_PARAMETER (TreeParam)
	xgboost::tree::DMLC_REGISTER_PARAMETER (TrainParam)
	dmlc::DMLC_REGISTRY_ENABLE (::xgboost::TreeGenReg)
std::vector< bst_cat_t >	xgboost::GetSplitCategories (RegTree const &tree, int32_t nidx)
std::string	xgboost::PrintCatsAsSet (std::vector< bst_cat_t > const &cats)
	xgboost::describe ("Dump text representation of tree") .set_body([](FeatureMap const &fmap
	xgboost::describe ("Dump json representation of tree") .set_body([](FeatureMap const &fmap
	xgboost::DMLC_REGISTER_PARAMETER (GraphvizParam)
	xgboost::describe ("Dump graphviz representation of tree") .set_body([](FeatureMap const &fmap
template<bool typed, bool feature_is_64>
void	xgboost::LoadModelImpl (Json const &in, TreeParam const &param, std::vector< RTreeNodeStat > *p_stats, std::vector< RegTree::Node > *p_nodes)

Variables
std::string const bool	xgboost::with_stats
std::string	xgboost::attrs

Detailed Description

Copyright 2015-2023 by Contributors.

model structure for tree

Macro Definition Documentation

XGBOOST_REGISTER_TREE_IO

#define XGBOOST_REGISTER_TREE_IO	(		UniqueId,
			Name
	)

Value:

  static DMLC_ATTRIBUTE_UNUSED ::xgboost::TreeGenReg&                   \
  __make_ ## TreeGenReg ## _ ## UniqueId ## __ =                        \
                  ::dmlc::Registry< ::xgboost::TreeGenReg>::Get()->__REGISTER__(Name)