Treelite
ast_native.h
Go to the documentation of this file.
1 
8 #ifndef TREELITE_COMPILER_AST_NATIVE_H_
9 #define TREELITE_COMPILER_AST_NATIVE_H_
10 
11 #include <treelite/compiler.h>
13 #include <memory>
14 
15 namespace treelite {
16 namespace compiler {
17 
18 class ASTNativeCompilerImpl;
19 
20 class ASTNativeCompiler : public Compiler {
21  public:
22  explicit ASTNativeCompiler(const CompilerParam& param);
23  virtual ~ASTNativeCompiler();
24  CompiledModel Compile(const Model& model) override;
25  CompilerParam QueryParam() const override;
26  private:
27  std::unique_ptr<ASTNativeCompilerImpl> pimpl_;
28 };
29 
30 } // namespace compiler
31 } // namespace treelite
32 
33 #endif // TREELITE_COMPILER_AST_NATIVE_H_
Parameters for tree compiler.
parameters for tree compiler
interface of compiler
Definition: compiler.h:53
Interface of compiler that compiles a tree ensemble model.
CompiledModel Compile(const Model &model) override
convert tree ensemble model
Definition: ast_native.cc:722
thin wrapper for tree ensemble model
Definition: tree.h:655
CompilerParam QueryParam() const override
Query the parameters used to intiailize the compiler.
Definition: ast_native.cc:727