treelite
Functions
Annotator

Functions

int TreeliteAnnotateBranch (ModelHandle model, DMatrixHandle dmat, int nthread, int verbose, AnnotationHandle *out)
 annotate branches in a given model using frequency patterns in the training data. More...
 
int TreeliteAnnotationSave (AnnotationHandle handle, const char *path)
 save branch annotation to a JSON file More...
 
int TreeliteAnnotationFree (AnnotationHandle handle)
 delete branch annotation from memory More...
 

Detailed Description

Branch annotator interface

Function Documentation

int TreeliteAnnotateBranch ( ModelHandle  model,
DMatrixHandle  dmat,
int  nthread,
int  verbose,
AnnotationHandle out 
)

annotate branches in a given model using frequency patterns in the training data.

Parameters
modelmodel to annotate
dmattraining data matrix
nthreadnumber of threads to use
verbosewhether to produce extra messages
outused to save handle for the created annotation
Returns
0 for success, -1 for failure

Definition at line 203 of file c_api.cc.

int TreeliteAnnotationFree ( AnnotationHandle  handle)

delete branch annotation from memory

Parameters
handleannotation to remove
Returns
0 for success, -1 for failure

Definition at line 226 of file c_api.cc.

int TreeliteAnnotationSave ( AnnotationHandle  handle,
const char *  path 
)

save branch annotation to a JSON file

Parameters
handleannotation to save
pathpath to JSON file
Returns
0 for success, -1 for failure

Definition at line 217 of file c_api.cc.