Treelite
Public Types | Public Attributes | List of all members
TaskParam Struct Reference

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...
 

Detailed Description

Group of parameters that are dependent on the choice of the task type.

Definition at line 174 of file tree.h.

Member Data Documentation

◆ grove_per_class

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.

Definition at line 185 of file tree.h.

◆ leaf_vector_size

unsigned int leaf_vector_size

Dimension of the output from each leaf node.

If >1, each leaf node produces a 1D vector output. If =1, each leaf node produces a single scalar.

Definition at line 200 of file tree.h.

◆ num_class

unsigned int num_class

The number of classes in the target label.

The num_class field should be >1 only when we're performing multi-class classification. Otherwise, for tasks such as binary classification, regression, and learning-to-rank, set num_class=1.

Definition at line 193 of file tree.h.


The documentation for this struct was generated from the following file: