OfflineLearner.hh

#ifndef GRAIL_OFFLINE_LEARNER_H
#define GRAIL_OFFLINE_LEARNER_H
 
#include <unordered_map>
#include <memory>
#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