treelite
|
This is the complete list of members for treelite::model_builder::ModelBuilder, including all inherited members.
CategoricalTest(std::int32_t split_index, bool default_left, std::vector< std::uint32_t > const &category_list, bool category_list_right_child, int left_child_key, int right_child_key)=0 | treelite::model_builder::ModelBuilder | pure virtual |
CommitModel()=0 | treelite::model_builder::ModelBuilder | pure virtual |
DataCount(std::uint64_t data_count)=0 | treelite::model_builder::ModelBuilder | pure virtual |
EndNode()=0 | treelite::model_builder::ModelBuilder | pure virtual |
EndTree()=0 | treelite::model_builder::ModelBuilder | pure virtual |
Gain(double gain)=0 | treelite::model_builder::ModelBuilder | pure virtual |
InitializeMetadata(Metadata const &metadata, TreeAnnotation const &tree_annotation, PostProcessorFunc const &postprocessor, std::vector< double > const &base_scores, std::optional< std::string > const &attributes)=0 | treelite::model_builder::ModelBuilder | pure virtual |
LeafScalar(double leaf_value)=0 | treelite::model_builder::ModelBuilder | pure virtual |
LeafVector(std::vector< float > const &leaf_vector)=0 | treelite::model_builder::ModelBuilder | pure virtual |
LeafVector(std::vector< double > const &leaf_vector)=0 | treelite::model_builder::ModelBuilder | pure virtual |
NumericalTest(std::int32_t split_index, double threshold, bool default_left, Operator cmp, int left_child_key, int right_child_key)=0 | treelite::model_builder::ModelBuilder | pure virtual |
StartNode(int node_key)=0 | treelite::model_builder::ModelBuilder | pure virtual |
StartTree()=0 | treelite::model_builder::ModelBuilder | pure virtual |
SumHess(double sum_hess)=0 | treelite::model_builder::ModelBuilder | pure virtual |
~ModelBuilder()=default | treelite::model_builder::ModelBuilder | virtual |