treelite
pred_transform.h
Go to the documentation of this file.
1 
7 #ifndef TREELITE_PRED_TRANSFORM_H_
8 #define TREELITE_PRED_TRANSFORM_H_
9 
10 #include <vector>
11 #include <string>
12 
13 namespace treelite {
14 namespace compiler {
15 
16 inline std::string PredTransformPrototype(bool batch) {
17  if (batch) {
18  return "size_t pred_transform_batch(float* pred, int64_t ndata, int nthread)";
19  } else {
20  return "size_t pred_transform(float* pred)";
21  }
22 }
23 
24 std::vector<std::string> PredTransformFunction(const Model& model, bool batch);
25 
26 } // namespace compiler
27 } // namespace treelite
28 
29 #endif // TREELITE_PRED_TRANSFORM_H_