C45Algorithm.h

#ifndef GRAIL_C45_ALGORITHM_H
#define GRAIL_C45_ALGORITHM_H
 
#include "../DT/DecisionNode.h"
#include "C45IterationParameters.h"
#include "C45ValueDecisionCounter.h"
#include "Dataset.hh"
#include <unordered_set>
#include <memory>
 
namespace grail
{
 namespace simulation
 { 
 class ISimulatedGameAction;
 
 class C45Algorithm
 {
 public:
 std::unique_ptr<DecisionNode<ISimulatedGameAction>> ConstructTree(Dataset& dataset, int maxDepth = decisionTreeMaxDepth);
 
 const std::unordered_set<int>& GetUsedColumns() const;
 
 private:
 void RunForContinuousColumn(size_t column, double globalEntropy, C45IterationParameters& params);
 void RunForNominalColumn(size_t column, double globalEntropy, C45IterationParameters& params);
 std::unique_ptr<DecisionNode<ISimulatedGameAction>> Iterate(C45IterationParameters params);
 std::unique_ptr<DecisionNode<ISimulatedGameAction>> PerformSplit(C45IterationParameters& params, int bestColumnIndex);
 
 const ISimulatedGameAction* globalFrequentAction = nullptr;
 std::unordered_set<int> usedColumns;
 int maxDepth;
 int currentDepth;
 
 };
 }
}
 
#endif //GRAIL_C45_ALGORITHM_H