DecisionNode.h

#ifndef GRAIL_DECISION_NODE_H
#define GRAIL_DECISION_NODE_H
 
#include <memory>
#include <string>
#include <unordered_map>
#include <vector>
#include "DecisionNodeType.h"
 
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