Treelite
|
Group of parameters that are dependent on the choice of the task type. More...
#include <tree.h>
Public Types | |
enum | OutputType : uint8_t { kFloat = 0, kInt = 1 } |
Public Attributes | |
OutputType | output_type |
The type of output from each leaf node. | |
bool | grove_per_class |
Whether we designate a subset of the trees to compute the prediction for each class. More... | |
unsigned int | num_class |
The number of classes in the target label. More... | |
unsigned int | leaf_vector_size |
Dimension of the output from each leaf node. More... | |
Group of parameters that are dependent on the choice of the task type.
bool grove_per_class |
Whether we designate a subset of the trees to compute the prediction for each class.
If True, the prediction for the i-th class is determined by the trees whose index is congruent to [i] modulo [num_class]. Only applicable if we are performing classification task with num_class > 2.
unsigned int leaf_vector_size |
unsigned int num_class |