Treelite
pred_transform.h
1 
8 #ifndef TREELITE_GTIL_PRED_TRANSFORM_H_
9 #define TREELITE_GTIL_PRED_TRANSFORM_H_
10 
11 #include <string>
12 #include <cstddef>
13 
14 namespace treelite {
15 
16 class Model;
17 
18 namespace gtil {
19 
20 using PredTransformFuncType = std::size_t (*) (const treelite::Model&, const float*, float*);
21 
22 PredTransformFuncType LookupPredTransform(const std::string& name);
23 
24 } // namespace gtil
25 } // namespace treelite
26 
27 #endif // TREELITE_GTIL_PRED_TRANSFORM_H_
thin wrapper for tree ensemble model
Definition: tree.h:717