treelite
base.h
Go to the documentation of this file.
1 
7 #ifndef TREELITE_BASE_H_
8 #define TREELITE_BASE_H_
9 
10 #include <string>
11 #include <unordered_map>
12 #include <cstdint>
13 
14 namespace treelite {
15 
17 typedef float tl_float;
19 enum class SplitFeatureType : int8_t {
20  kNone, kNumerical, kCategorical
21 };
23 enum class Operator : int8_t {
24  kEQ,
25  kLT,
26  kLE,
27  kGT,
28  kGE
29 };
31 extern const std::unordered_map<std::string, Operator> optable;
32 
33 } // namespace treelite
34 
35 #endif // TREELITE_BASE_H_
SplitFeatureType
feature split type
Definition: base.h:19
float tl_float
float type to be used internally
Definition: base.h:17
const std::unordered_map< std::string, Operator > optable
conversion table from string to operator, defined in optable.cc
Definition: optable.cc:12
Operator
comparison operators
Definition: base.h:23