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, const Model& model);
18 
19 } // namespace compiler
20 } // namespace treelite
21 
22 #endif // TREELITE_COMPILER_PRED_TRANSFORM_H_
model structure for tree ensemble