7 #ifndef TREELITE_BASE_H_ 8 #define TREELITE_BASE_H_ 11 #include <unordered_map> 20 kNone, kNumerical, kCategorical
31 extern const std::unordered_map<std::string, Operator>
optable;
40 case Operator::kEQ:
return "==";
41 case Operator::kLT:
return "<";
42 case Operator::kLE:
return "<=";
43 case Operator::kGT:
return ">";
44 case Operator::kGE:
return ">=";
51 #endif // TREELITE_BASE_H_ SplitFeatureType
feature split type
std::string OpName(Operator op)
get string representation of comparsion operator
const std::unordered_map< std::string, Operator > optable
conversion table from string to operator, defined in optable.cc
double tl_float
float type to be used internally
Operator
comparison operators