7 #ifndef TREELITE_COMPILER_H_ 8 #define TREELITE_COMPILER_H_ 10 #include <dmlc/registry.h> 25 std::unordered_map<std::string, std::string> files;
26 std::string file_prefix;
46 static Compiler* Create(
const std::string& name,
54 :
public dmlc::FunctionRegEntryBase<CompilerReg,
55 std::function<Compiler* (const compiler::CompilerParam&)> > {
70 #define TREELITE_REGISTER_COMPILER(UniqueId, Name) \ 71 static DMLC_ATTRIBUTE_UNUSED ::treelite::CompilerReg & \ 72 __make_ ## CompilerReg ## _ ## UniqueId ## __ = \ 73 ::dmlc::Registry< ::treelite::CompilerReg>::Get()->__REGISTER__(Name) 77 #endif // TREELITE_COMPILER_H_
thin wrapper for tree ensemble model
parameters for tree compiler
Registry entry for compiler.