Metadata object, consisting of metadata information about the model at large. More...

Collaboration diagram for treelite::model_builder::Metadata:

Public Member Functions
	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. More...

Public Attributes
std::int32_t	num_feature {0}

TaskType	task_type {TaskType::kRegressor}

bool	average_tree_output {false}

std::int32_t	num_target {1}

std::vector< std::int32_t >	num_class {1}

std::array< std::int32_t, 2 >	leaf_vector_shape {1, 1}

Detailed Description

Metadata object, consisting of metadata information about the model at large.

Constructor & Destructor Documentation

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

bool treelite::model_builder::Metadata::average_tree_output {false}

std::array<std::int32_t, 2> treelite::model_builder::Metadata::leaf_vector_shape {1, 1}

std::vector<std::int32_t> treelite::model_builder::Metadata::num_class {1}

std::int32_t treelite::model_builder::Metadata::num_feature {0}

std::int32_t treelite::model_builder::Metadata::num_target {1}

TaskType treelite::model_builder::Metadata::task_type {TaskType::kRegressor}

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