treelite
Public Member Functions | Public Attributes | List of all members
Model Struct Reference

thin wrapper for tree ensemble model More...

#include <tree.h>

Collaboration diagram for Model:
Collaboration graph
[legend]

Public Member Functions

 Model ()
 disable copy; use default move
 
 Model (const Model &)=delete
 
Modeloperator= (const Model &)=delete
 
 Model (Model &&)=default
 
Modeloperator= (Model &&)=default
 

Public Attributes

std::vector< Treetrees
 member trees
 
int num_feature
 number of features used for the model. It is assumed that all feature indices are between 0 and [num_feature]-1.
 
int num_output_group
 number of output groups – for multi-class classification Set to 1 for everything else
 
bool random_forest_flag
 flag for random forest; True for random forests and False for gradient boosted trees
 
ModelParam param
 extra parameters
 

Detailed Description

thin wrapper for tree ensemble model

Definition at line 415 of file tree.h.


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