7 #ifndef TREELITE_PARAM_H_ 8 #define TREELITE_PARAM_H_ 10 #include <dmlc/parameter.h> 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");
55 #endif // TREELITE_PARAM_H_ parameters for tree compiler
std::string annotate_in
name of model annotation file. Use the class treelite.Annotator to generate this file.
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