treelite
treelite::model_builder::ModelBuilder Member List

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)=0treelite::model_builder::ModelBuilderpure virtual
CommitModel()=0treelite::model_builder::ModelBuilderpure virtual
DataCount(std::uint64_t data_count)=0treelite::model_builder::ModelBuilderpure virtual
EndNode()=0treelite::model_builder::ModelBuilderpure virtual
EndTree()=0treelite::model_builder::ModelBuilderpure virtual
Gain(double gain)=0treelite::model_builder::ModelBuilderpure virtual
InitializeMetadata(Metadata const &metadata, TreeAnnotation const &tree_annotation, PostProcessorFunc const &postprocessor, std::vector< double > const &base_scores, std::optional< std::string > const &attributes)=0treelite::model_builder::ModelBuilderpure virtual
LeafScalar(double leaf_value)=0treelite::model_builder::ModelBuilderpure virtual
LeafVector(std::vector< float > const &leaf_vector)=0treelite::model_builder::ModelBuilderpure virtual
LeafVector(std::vector< double > const &leaf_vector)=0treelite::model_builder::ModelBuilderpure virtual
NumericalTest(std::int32_t split_index, double threshold, bool default_left, Operator cmp, int left_child_key, int right_child_key)=0treelite::model_builder::ModelBuilderpure virtual
StartNode(int node_key)=0treelite::model_builder::ModelBuilderpure virtual
StartTree()=0treelite::model_builder::ModelBuilderpure virtual
SumHess(double sum_hess)=0treelite::model_builder::ModelBuilderpure virtual
~ModelBuilder()=defaulttreelite::model_builder::ModelBuildervirtual