Treelite
Public Member Functions | Public Attributes | List of all members
ModelBuilderImpl Struct Reference

Public Member Functions

 ModelBuilderImpl (int num_feature, int num_class, bool average_tree_output, TypeInfo threshold_type, TypeInfo leaf_output_type)
 
template<typename ThresholdType , typename LeafOutputType >
void CommitModelImpl (ModelImpl< ThresholdType, LeafOutputType > *out_model)
 

Public Attributes

std::vector< TreeBuildertrees
 
int num_feature
 
int num_class
 
bool average_tree_output
 
TypeInfo threshold_type
 
TypeInfo leaf_output_type
 
std::vector< std::pair< std::string, std::string > > cfg
 

Detailed Description

Definition at line 76 of file builder.cc.


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