DecisionNode.h

// Copyright QED Software 2023.
 
#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 simgames
{
namespace dt
{
 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