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