1 #ifndef GRAIL_DECISION_NODE_H
2 #define GRAIL_DECISION_NODE_H
6 #include <unordered_map>
8 #include "DecisionNodeType.h"
14 const int decisionTreeMaxDepth = 99999;
19 template <
class TDecisionType>
26 const virtual TDecisionType* Predict(std::vector<float>& data)
const = 0;
27 virtual void Print(std::unordered_map<int, std::string>& columnNames,
28 const std::string& indent =
"")
const = 0;
29 virtual DecisionNodeType GetNodeType()
const = 0;
35 #endif //GRAIL_DECISION_NODE_H