Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions | Static Public Member Functions | Protected Member Functions | Static Protected Member Functions | Protected Attributes | Static Protected Attributes
xgboost::TreeGenerator Class Referenceabstract

Base class for dump model implementation, modeling closely after code generator. More...

Inheritance diagram for xgboost::TreeGenerator:
xgboost::GraphvizGenerator xgboost::JsonGenerator xgboost::TextGenerator

Public Member Functions

 TreeGenerator (FeatureMap const &_fmap, bool with_stats)
 
virtual void BuildTree (RegTree const &tree)
 
std::string Str () const
 

Static Public Member Functions

static TreeGeneratorCreate (std::string const &attrs, FeatureMap const &fmap, bool with_stats)
 

Protected Member Functions

virtual std::string Indicator (RegTree const &, int32_t, uint32_t) const
 
virtual std::string Categorical (RegTree const &, int32_t, uint32_t) const =0
 
virtual std::string Integer (RegTree const &, int32_t, uint32_t) const
 
virtual std::string Quantitive (RegTree const &, int32_t, uint32_t) const
 
virtual std::string NodeStat (RegTree const &, int32_t) const
 
virtual std::string PlainNode (RegTree const &, int32_t, uint32_t) const =0
 
virtual std::string SplitNode (RegTree const &tree, int32_t nid, uint32_t depth)
 
virtual std::string LeafNode (RegTree const &tree, int32_t nid, uint32_t depth) const =0
 
virtual std::string BuildTree (RegTree const &tree, int32_t nid, uint32_t depth)=0
 

Static Protected Member Functions

template<typename Float >
static std::string ToStr (Float value)
 
static std::string Tabs (uint32_t n)
 
static std::string Match (std::string const &input, std::map< std::string, std::string > const &replacements)
 

Protected Attributes

FeatureMap const & fmap_
 
std::stringstream ss_
 
bool const with_stats_
 

Static Protected Attributes

static int32_t constexpr kFloatMaxPrecision
 

Detailed Description

Base class for dump model implementation, modeling closely after code generator.

Field Documentation

◆ kFloatMaxPrecision

int32_t constexpr xgboost::TreeGenerator::kFloatMaxPrecision
staticconstexprprotected
Initial value:
=
std::numeric_limits<bst_float>::max_digits10

The documentation for this class was generated from the following file: