DecisionNode.h

#ifndef GRAIL_DECISION_NODE_H
#define GRAIL_DECISION_NODE_H
 
#include "DecisionNodeType.h"
#include <memory>
#include <vector>
#include <unordered_map>
#include <string>
 
namespace grail
{
 namespace simulation
 {
 const int decisionTreeMaxDepth = 99999;
 
 template <class TDecisionType>
 struct DecisionNode
 {
 virtual ~DecisionNode()
 {
 
 }
 
 const virtual TDecisionType* Predict(std::vector<float>& data) const = 0; 
 virtual void Print(std::unordered_map<int, std::string>& columnNames, const std::string& indent = "") const = 0;
 virtual DecisionNodeType GetNodeType() const = 0;
 template <class T> friend struct IDecisionTreeSerializer;
 };
 }
}
#endif //GRAIL_DECISION_NODE_H