Treelite
|
thin wrapper for tree ensemble model More...
#include <tree.h>
Public Member Functions | |
Model ()=default | |
disable copy; use default move | |
Model (const Model &)=delete | |
Model & | operator= (const Model &)=delete |
Model (Model &&)=default | |
Model & | operator= (Model &&)=default |
void | ReferenceSerialize (dmlc::Stream *fo) const |
std::vector< PyBufferFrame > | GetPyBuffer () |
void | InitFromPyBuffer (std::vector< PyBufferFrame > frames) |
Model | Clone () const |
Public Attributes | |
std::vector< Tree > | trees |
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 | |