C45GameActionCounter.h

#ifndef GRAIL_C45_GAME_ACTION_COUNTER_H
#define GRAIL_C45_GAME_ACTION_COUNTER_H
 
#include "../ISimulatedGameAction.hh"
#include <unordered_map>
#include <cmath>
 
namespace grail
{
    namespace simulation
    {   
        class C45GameActionCounter
        {
        public:
            void clear();
 
            double getTotalSampleCount() const;
 
            size_t geKeyCount() const;
 
            const ISimulatedGameAction* frequentValue() const;
 
            void observeKey(const ISimulatedGameAction* key);
    
            double calculateEntropy() const;
 
            int getCount(const ISimulatedGameAction* key) const;
 
        private:
            std::unordered_map<const ISimulatedGameAction*, int, HashForOfflineLearning, EqualsForOfflineLearning> sampleCountByKey;
            double totalSamples = 0;
        };
    }
}
 
#endif //GRAIL_C45_GAME_ACTION_COUNTER_H