Treelite
pred_transform.h
1 
7 #ifndef TREELITE_COMPILER_PRED_TRANSFORM_H_
8 #define TREELITE_COMPILER_PRED_TRANSFORM_H_
9 
10 #include <treelite/tree.h>
11 #include <vector>
12 #include <string>
13 
14 namespace treelite {
15 namespace compiler {
16 
17 std::string PredTransformFunction(const std::string& backend,
18  const Model& model);
19 
20 } // namespace compiler
21 } // namespace treelite
22 
23 #endif // TREELITE_COMPILER_PRED_TRANSFORM_H_
model structure for tree ensemble