tree_node_type.h

Go to the documentation of this file.

 
#ifndef TREELITE_ENUM_TREE_NODE_TYPE_H_
#define TREELITE_ENUM_TREE_NODE_TYPE_H_
 
#include <cstdint>
#include <string>
 
namespace treelite {
 
enum class TreeNodeType : std::int8_t {
  kLeafNode = 0,
  kNumericalTestNode = 1,
  kCategoricalTestNode = 2
};
 
std::string TreeNodeTypeToString(TreeNodeType type);
 
TreeNodeType TreeNodeTypeFromString(std::string const& name);
 
}  // namespace treelite
 
#endif  // TREELITE_ENUM_TREE_NODE_TYPE_H_