Treelite
Functions
Compiler

Functions

int TreeliteCompilerCreateV2 (const char *name, const char *params_json_str, CompilerHandle *out)
 Create a compiler with a given name. More...
 
int TreeliteCompilerGenerateCodeV2 (CompilerHandle compiler, ModelHandle model, const char *dirpath)
 Generate prediction code from a tree ensemble model. The code will be C99 compliant. One header file (.h) will be generated, along with one or more source files (.c). More...
 
int TreeliteCompilerFree (CompilerHandle handle)
 delete compiler from memory More...
 

Detailed Description

Compiler interface

Function Documentation

◆ TreeliteCompilerCreateV2()

int TreeliteCompilerCreateV2 ( const char *  name,
const char *  params_json_str,
CompilerHandle out 
)

Create a compiler with a given name.

Parameters
namename of compiler
params_json_strJSON string representing the parameters for the compiler
outcreated compiler
Returns
0 for success, -1 for failure

Definition at line 54 of file c_api.cc.

◆ TreeliteCompilerFree()

int TreeliteCompilerFree ( CompilerHandle  handle)

delete compiler from memory

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

Definition at line 96 of file c_api.cc.

◆ TreeliteCompilerGenerateCodeV2()

int TreeliteCompilerGenerateCodeV2 ( CompilerHandle  compiler,
ModelHandle  model,
const char *  dirpath 
)

Generate prediction code from a tree ensemble model. The code will be C99 compliant. One header file (.h) will be generated, along with one or more source files (.c).

Usage example:

TreeliteCompilerGenerateCodeV2(compiler, model, "./my/model");
// files to generate: ./my/model/header.h, ./my/model/main.c
// if parallel compilation is enabled:
// ./my/model/header.h, ./my/model/main.c, ./my/model/tu0.c,
// ./my/model/tu1.c, and so forth
Parameters
compilerhandle for compiler
modelhandle for tree ensemble model
dirpathdirectory to store header and source files
Returns
0 for success, -1 for failure

Definition at line 61 of file c_api.cc.