Metadata object, consisting of metadata information about the model at large.
More...
#include <model_builder.h>
Metadata object, consisting of metadata information about the model at large.
◆ Metadata()
treelite::model_builder::Metadata::Metadata |
( |
std::int32_t |
num_feature, |
|
|
TaskType |
task_type, |
|
|
bool |
average_tree_output, |
|
|
std::int32_t |
num_target, |
|
|
std::vector< std::int32_t > const & |
num_class, |
|
|
std::array< std::int32_t, 2 > const & |
leaf_vector_shape |
|
) |
| |
Constructor for Metadata object.
- Parameters
-
num_feature | Number of features |
task_type | Task type |
average_tree_output | Whether to average outputs of trees |
num_target | Number of targets |
num_class | Number of classes. num_class[i] is the number of classes of target i. |
leaf_vector_shape | Shape of the output from each leaf node |
◆ average_tree_output
bool treelite::model_builder::Metadata::average_tree_output {false} |
◆ leaf_vector_shape
std::array<std::int32_t, 2> treelite::model_builder::Metadata::leaf_vector_shape {1, 1} |
◆ num_class
std::vector<std::int32_t> treelite::model_builder::Metadata::num_class {1} |
◆ num_feature
std::int32_t treelite::model_builder::Metadata::num_feature {0} |
◆ num_target
std::int32_t treelite::model_builder::Metadata::num_target {1} |
◆ task_type
The documentation for this struct was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/treelite/checkouts/latest/include/treelite/model_builder.h