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