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 (std::istream &fi) |
load branch annotation from a JSON file More... | |
void | Save (std::ostream &fo) const |
save branch annotation to a JSON file More... | |
std::vector< std::vector< uint64_t > > | Get () const |
fetch branch annotation. Usage example: More... | |
branch annotator class
Definition at line 21 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 237 of file annotator.cc.
|
inline |
fetch branch annotation. Usage example:
Definition at line 55 of file annotator.h.
void Load | ( | std::istream & | fi | ) |
load branch annotation from a JSON file
fi | input stream |
Definition at line 245 of file annotator.cc.
void Save | ( | std::ostream & | fo | ) | const |
save branch annotation to a JSON file
fo | output stream |
Definition at line 264 of file annotator.cc.