Treelite
compiler_param.h
Go to the documentation of this file.
1 
7 #ifndef TREELITE_COMPILER_PARAM_H_
8 #define TREELITE_COMPILER_PARAM_H_
9 
10 #include <string>
11 #include <limits>
12 
13 namespace treelite {
14 namespace compiler {
15 
17 struct CompilerParam {
25  std::string annotate_in;
27  int quantize;
35  int verbose;
37  std::string native_lib_name;
51  static CompilerParam ParseFromJSON(const char* param_json_str);
52 };
53 
54 } // namespace compiler
55 } // namespace treelite
56 
57 #endif // TREELITE_COMPILER_PARAM_H_
parameters for tree compiler
std::string native_lib_name
native lib name (without extension)
std::string annotate_in
name of model annotation file. Use the class treelite.Annotator to generate this file.
int dump_array_as_elf
Only applicable when compiler is set to failsafe. If set to a positive value, the fail-safe compiler ...
double code_folding_req
parameter for folding rarely visited subtrees (no if/else blocks); all nodes whose data counts are lo...
int parallel_comp
option to enable parallel compilation; if set to nonzero, the trees will be evely distributed into [p...
int quantize
whether to quantize threshold points (0: no, >0: yes)
int verbose
if >0, produce extra messages