DecisionTree.hh

#ifndef GRAIL_DECISION_TREE_H
#define GRAIL_DECISION_TREE_H
 
#include "../ISimulatedGameHeuristic.hh"
#include "DecisionNode.h"
#include <unordered_set>
#include <unordered_map>
#include <initializer_list>
 
namespace grail
{
 namespace simulation
 {
 class ISimulatedGameAction;
 struct IVectorizer;
 class Dataset;
 
 template <class T>
 struct IDecisionTreeSerializer;
 
 class DecisionTree : public SimulatedGameHeuristic
 {
 public:
 DecisionTree(std::unique_ptr<IVectorizer> vectorizer);
 
 void Print();
 
 void Serialize(IDecisionTreeSerializer<ISimulatedGameAction>& serializer) const;
 
 void Deserialize(IDecisionTreeSerializer<ISimulatedGameAction>& serializer);
 
 void Construct(Dataset& dataset, int maxDepth = decisionTreeMaxDepth);
 
 virtual std::unique_ptr<ISimulatedGameAction> GetAction() const override;
 
 virtual bool IsHeuristicSituation(const ISimulatedGameUnit& unit) const final;
 
 void SetColumnName(int columnIndex, const std::string& name);
 
 void SetColumnNames(const std::initializer_list<std::string>& consecutiveNames);
 
 DecisionTree(const DecisionTree& other) = delete;
 
 IVectorizer& GetVectorizer() const;
 
 void SetVectorizer(std::unique_ptr<IVectorizer> vectorizer);
 private:
 void ConstructDefaultColumnNames();
 
 std::unique_ptr<DecisionNode<ISimulatedGameAction>> root;
 std::unique_ptr<IVectorizer> vectorizer;
 std::unordered_set<int> usedConsiderationIndices;
 std::unordered_map<int, std::string> columnNames;
 };
 }
}
 
#endif //GRAIL_DECISION_TREE_H