treelite
Public Member Functions | Public Attributes | List of all members
CompilerParam Struct Reference

parameters for tree compiler More...

#include <param.h>

Inheritance diagram for CompilerParam:
Inheritance graph
[legend]
Collaboration diagram for CompilerParam:
Collaboration graph
[legend]

Public Member Functions

 DMLC_DECLARE_PARAMETER (CompilerParam)
 

Public Attributes

std::string annotate_in
 name of model annotation file. Use the class treelite.Annotator to generate this file.
 
int quantize
 whether to quantize threshold points (0: no, >0: yes)
 
int parallel_comp
 option to enable parallel compilation; if set to nonzero, the trees will be evely distributed into [parallel_comp] files. Set this option to improve compilation time and reduce memory consumption during compilation.
 
int verbose
 if >0, produce extra messages
 
std::string native_lib_name
 native lib name (without extension)
 
double code_folding_req
 parameter for folding rarely visited subtrees (no if/else blocks); all nodes whose data counts are lower than that of the root node of the decision tree by [code_folding_req] will be folded. To diable folding, set to +inf. If hessian sums are available, they will be used as proxies of data counts.
 

Detailed Description

parameters for tree compiler

Definition at line 18 of file param.h.


The documentation for this struct was generated from the following file: