treelite
param.h
Go to the documentation of this file.
1 
7 #ifndef TREELITE_PARAM_H_
8 #define TREELITE_PARAM_H_
9 
10 #include <dmlc/parameter.h>
11 
12 namespace treelite {
13 namespace compiler {
14 
16 struct CompilerParam : public dmlc::Parameter<CompilerParam> {
24  std::string annotate_in;
26  int quantize;
34  int verbose;
37  // declare parameters
38  DMLC_DECLARE_PARAMETER(CompilerParam) {
39  DMLC_DECLARE_FIELD(annotate_in).set_default("NULL")
40  .describe("Name of model annotation file");
41  DMLC_DECLARE_FIELD(quantize).set_lower_bound(0).set_default(0)
42  .describe("whether to quantize threshold points (0: no, >0: yes)");
43  DMLC_DECLARE_FIELD(parallel_comp).set_lower_bound(0).set_default(0)
44  .describe("option to enable parallel compilation;"
45  "if set to nonzero, the trees will be evely distributed"
46  "into [parallel_comp] files.");
47  DMLC_DECLARE_FIELD(verbose).set_default(0)
48  .describe("if >0, produce extra messages");
49  }
50 };
51 
52 } // namespace compiler
53 } // namespace treelite
54 
55 #endif // TREELITE_PARAM_H_
parameters for tree compiler
Definition: param.h:16
std::string annotate_in
name of model annotation file. Use the class treelite.Annotator to generate this file.
Definition: param.h:24
int parallel_comp
option to enable parallel compilation; if set to nonzero, the trees will be evely distributed into [p...
Definition: param.h:32
int quantize
whether to quantize threshold points (0: no, >0: yes)
Definition: param.h:26
int verbose
if >0, produce extra messages
Definition: param.h:34