Treelite
Public Member Functions | Friends | List of all members
ASTBuilder< ThresholdType, LeafOutputType > Class Template Reference

Public Member Functions

void BuildAST (const ModelImpl< ThresholdType, LeafOutputType > &model)
 
std::vector< bool > GenerateIsCategoricalArray ()
 
bool FoldCode (double magnitude_req, bool create_new_translation_unit=false)
 
void Split (int parallel_comp)
 
void QuantizeThresholds ()
 
void LoadDataCounts (const std::vector< std::vector< uint64_t >> &counts)
 
std::string GetDump () const
 
const ASTNodeGetRootNode ()
 

Friends

bool treelite::compiler::fold_code (ASTNode *, CodeFoldingContext *, ASTBuilder< ThresholdType, LeafOutputType > *)
 

Detailed Description

template<typename ThresholdType, typename LeafOutputType>
class treelite::compiler::ASTBuilder< ThresholdType, LeafOutputType >

Definition at line 24 of file builder.h.


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