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