Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions | Protected Member Functions
xgboost::GraphvizGenerator Class Reference
Inheritance diagram for xgboost::GraphvizGenerator:
xgboost::TreeGenerator

Public Member Functions

 GraphvizGenerator (FeatureMap const &fmap, std::string const &attrs, bool with_stats)
 
- Public Member Functions inherited from xgboost::TreeGenerator
 TreeGenerator (FeatureMap const &_fmap, bool with_stats)
 
std::string Str () const
 

Protected Member Functions

template<bool is_categorical>
std::string BuildEdge (RegTree const &tree, bst_node_t nid, int32_t child, bool left) const
 
std::string PlainNode (RegTree const &tree, int32_t nid, uint32_t) const override
 
std::string Categorical (RegTree const &tree, int32_t nid, uint32_t) const override
 
std::string LeafNode (RegTree const &tree, int32_t nid, uint32_t) const override
 
std::string BuildTree (RegTree const &tree, int32_t nid, uint32_t depth) override
 
void BuildTree (RegTree const &tree) override
 
- Protected Member Functions inherited from xgboost::TreeGenerator
virtual std::string Indicator (RegTree const &, int32_t, uint32_t) const
 
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 SplitNode (RegTree const &tree, int32_t nid, uint32_t depth)
 

Additional Inherited Members

- Static Public Member Functions inherited from xgboost::TreeGenerator
static TreeGeneratorCreate (std::string const &attrs, FeatureMap const &fmap, bool with_stats)
 
- Static Protected Member Functions inherited from xgboost::TreeGenerator
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 inherited from xgboost::TreeGenerator
FeatureMap const & fmap_
 
std::stringstream ss_
 
bool const with_stats_
 
- Static Protected Attributes inherited from xgboost::TreeGenerator
static int32_t constexpr kFloatMaxPrecision
 

Member Function Documentation

◆ BuildTree() [1/2]

void xgboost::GraphvizGenerator::BuildTree ( RegTree const &  tree)
inlineoverrideprotectedvirtual

Reimplemented from xgboost::TreeGenerator.

◆ BuildTree() [2/2]

std::string xgboost::GraphvizGenerator::BuildTree ( RegTree const &  tree,
int32_t  nid,
uint32_t  depth 
)
inlineoverrideprotectedvirtual

◆ Categorical()

std::string xgboost::GraphvizGenerator::Categorical ( RegTree const &  tree,
int32_t  nid,
uint32_t   
) const
inlineoverrideprotectedvirtual

◆ LeafNode()

std::string xgboost::GraphvizGenerator::LeafNode ( RegTree const &  tree,
int32_t  nid,
uint32_t   
) const
inlineoverrideprotectedvirtual

◆ PlainNode()

std::string xgboost::GraphvizGenerator::PlainNode ( RegTree const &  tree,
int32_t  nid,
uint32_t   
) const
inlineoverrideprotectedvirtual

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