treelite
|
branch annotator class More...
#include <annotator.h>
Public Member Functions | |
void | Annotate (const Model &model, const DMatrix *dmat, int nthread, int verbose) |
annotate branches in a given model using frequency patterns in the training data. The annotation can be accessed through Get() method. More... | |
void | Load (dmlc::Stream *fi) |
load branch annotation from a JSON file More... | |
void | Save (dmlc::Stream *fo) const |
save branch annotation to a JSON file More... | |
std::vector< std::vector< size_t > > | Get () const |
fetch branch annotation. Usage example: More... | |
branch annotator class
Definition at line 16 of file annotator.h.
annotate branches in a given model using frequency patterns in the training data. The annotation can be accessed through Get() method.
model | tree ensemble model |
dmat | training data matrix |
nthread | number of threads to use |
verbose | whether to produce extra messages |
Definition at line 114 of file annotator.cc.
|
inline |
fetch branch annotation. Usage example:
Definition at line 51 of file annotator.h.
void Load | ( | dmlc::Stream * | fi | ) |
load branch annotation from a JSON file
fi | input stream |
Definition at line 157 of file annotator.cc.
void Save | ( | dmlc::Stream * | fo | ) | const |
save branch annotation to a JSON file
fo | output stream |
Definition at line 164 of file annotator.cc.