OfflineLearner.hh

#ifndef GRAIL_OFFLINE_LEARNER_H
#define GRAIL_OFFLINE_LEARNER_H
 
#include <memory>
#include <unordered_map>
#include <vector>
 
namespace grail
{
    namespace simulation
    {
        class SimulatedGameNode;
        struct IVectorizer;
        class UniqueTreeDataset;
        class ISimulatedGameUnit;
 
        class OfflineLearner
        {
        public:
            OfflineLearner(std::unique_ptr<IVectorizer> vectorizer);
 
            void ClearData();
 
            std::unique_ptr<UniqueTreeDataset> GetSamplesDataset(double minFractionOfSimulationsThreshold = 0.7);
 
            void FillSamplesDataset(UniqueTreeDataset& dataset, double minFractionOfSimulationsThreshold = 0.7);
 
        private:
            double GetSampleMaxVisits() const;
            void ObserveNode(const SimulatedGameNode* node);
            bool IsLearnableSituation(const ISimulatedGameUnit& unit);
 
            std::unique_ptr<IVectorizer> stateObserver;
            std::unordered_map<const SimulatedGameNode*, std::vector<float>> capturedData = {};
 
            friend class SimulatedGameThinkingUnit;
        };
    }
}
 
#endif //GRAIL_OFFLINE_LEARNER_H