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

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 428 of file tree.h.

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