7 #ifndef TREELITE_COMPILER_H_ 8 #define TREELITE_COMPILER_H_ 10 #include <dmlc/registry.h> 11 #include <unordered_map> 29 std::vector<char> content_binary;
35 explicit FileEntry(
const std::string& content) =
delete;
37 : content(std::move(content)), is_binary(
false) {}
38 explicit FileEntry(
const std::vector<char>&) =
delete;
39 explicit FileEntry(std::vector<char>&& content)
40 : content_binary(std::move(content)), is_binary(
true) {}
47 std::unordered_map<std::string, FileEntry> files;
48 std::string file_prefix;
68 static Compiler* Create(
const std::string& name,
76 :
public dmlc::FunctionRegEntryBase<CompilerReg,
77 std::function<Compiler* (const compiler::CompilerParam&)> > {
92 #define TREELITE_REGISTER_COMPILER(UniqueId, Name) \ 93 static DMLC_ATTRIBUTE_UNUSED ::treelite::CompilerReg & \ 94 __make_ ## CompilerReg ## _ ## UniqueId ## __ = \ 95 ::dmlc::Registry< ::treelite::CompilerReg>::Get()->__REGISTER__(Name) 99 #endif // TREELITE_COMPILER_H_
thin wrapper for tree ensemble model
parameters for tree compiler
Registry entry for compiler.