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