C45ValueDecisionCounter.h

#ifndef GRAIL_C45_VALUE_DECISION_COUNTER_H
#define GRAIL_C45_VALUE_DECISION_COUNTER_H
 
#include "C45GameActionCounter.h"
#include <unordered_map>
 
namespace grail
{
    namespace simulation
    {
        class ISimulatedGameAction;
 
        class C45ValueDecisionCounter 
        {
        public:
            
            const std::unordered_map<float, int>& getSampleCountByKey() const;
 
            void clear();
 
            double getTotalSampleCount() const;
 
            size_t geKeyCount() const;
 
            void observeKey(float key);
 
            void observeKeyDecision(float key, ISimulatedGameAction* decision);
 
            double calculateEntropy() const;
 
            int getCount(float key) const;
 
            double calculateGainPortion(double globalEntropy);
 
        private:
            std::unordered_map<float, int> sampleCountByKey;
            std::unordered_map<float, C45GameActionCounter> decisionCounterByKey;
            double totalSamples = 0;
        };
    }
}
 
#endif //GRAIL_C45_VALUE_DECISION_COUNTER_H