Model class for tree ensemble model.
More...
#include <tree.h>
Model class for tree ensemble model.
◆ Model() [1/3]
treelite::Model::Model |
( |
| ) |
|
|
inline |
disable copy; use default move
◆ ~Model()
virtual treelite::Model::~Model |
( |
| ) |
|
|
virtualdefault |
◆ Model() [2/3]
treelite::Model::Model |
( |
Model const & |
| ) |
|
|
delete |
◆ Model() [3/3]
treelite::Model::Model |
( |
Model && |
| ) |
|
|
default |
◆ Create() [1/2]
template<typename ThresholdType , typename LeafOutputType >
std::unique_ptr< Model > treelite::Model::Create |
|
inlinestatic |
◆ Create() [2/2]
◆ DeserializeFromPyBuffer()
◆ DeserializeFromStream()
static std::unique_ptr<Model> treelite::Model::DeserializeFromStream |
( |
std::istream & |
is | ) |
|
|
static |
◆ DumpAsJSON() [1/2]
std::string treelite::Model::DumpAsJSON |
( |
bool |
pretty_print | ) |
const |
|
inline |
◆ DumpAsJSON() [2/2]
void treelite::Model::DumpAsJSON |
( |
std::ostream & |
fo, |
|
|
bool |
pretty_print |
|
) |
| const |
◆ GetHeaderField()
PyBufferFrame treelite::Model::GetHeaderField |
( |
std::string const & |
name | ) |
|
Get a field in the header.
◆ GetLeafOutputType()
TypeInfo treelite::Model::GetLeafOutputType |
( |
| ) |
const |
|
inline |
◆ GetNumTree()
std::size_t treelite::Model::GetNumTree |
( |
| ) |
const |
|
inline |
◆ GetThresholdType()
TypeInfo treelite::Model::GetThresholdType |
( |
| ) |
const |
|
inline |
◆ GetTreeField()
PyBufferFrame treelite::Model::GetTreeField |
( |
std::uint64_t |
tree_id, |
|
|
std::string const & |
name |
|
) |
| |
◆ GetVersion()
Version treelite::Model::GetVersion |
( |
| ) |
const |
|
inline |
Return the Treelite version that produced this Model object.
◆ operator=() [1/2]
◆ operator=() [2/2]
Model& treelite::Model::operator= |
( |
Model const & |
| ) |
|
|
delete |
◆ SerializeToPyBuffer()
◆ SerializeToStream()
void treelite::Model::SerializeToStream |
( |
std::ostream & |
os | ) |
|
◆ SetHeaderField()
void treelite::Model::SetHeaderField |
( |
std::string const & |
name, |
|
|
PyBufferFrame |
frame |
|
) |
| |
Set a field in the header.
◆ SetTreeField()
void treelite::Model::SetTreeField |
( |
std::uint64_t |
tree_id, |
|
|
std::string const & |
name, |
|
|
PyBufferFrame |
frame |
|
) |
| |
◆ SetTreeLimit()
void treelite::Model::SetTreeLimit |
( |
std::size_t |
limit | ) |
|
|
inline |
◆ detail::serializer::Deserializer
template<typename MixIn >
◆ detail::serializer::Serializer
template<typename MixIn >
◆ attributes
std::string treelite::Model::attributes |
◆ average_tree_output
bool treelite::Model::average_tree_output {false} |
whether to average tree outputs
◆ base_scores
◆ class_id
◆ leaf_vector_shape
◆ num_class
◆ num_feature
std::int32_t treelite::Model::num_feature {0} |
Number of features used for the model. It is assumed that all feature indices are between 0 and [num_feature]-1.
◆ num_target
std::int32_t treelite::Model::num_target |
◆ postprocessor
std::string treelite::Model::postprocessor |
◆ ratio_c
float treelite::Model::ratio_c {1.0f} |
◆ sigmoid_alpha
float treelite::Model::sigmoid_alpha {1.0f} |
◆ target_id
◆ task_type
◆ variant_
The documentation for this class was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/treelite/checkouts/latest/include/treelite/tree.h