OfflineLearner.hh

// Copyright QED Software 2023.
 
#ifndef GRAIL_OFFLINE_LEARNER_H
#define GRAIL_OFFLINE_LEARNER_H
 
#include <memory>
#include <unordered_map>
#include <vector>
 
namespace grail
{
namespace simgames
{
 class ISimulatedGameUnit;
 class SimulatedGameNode;
 class SimulatedGameThinkingUnit;
 
namespace learn
{
 struct IVectorizer;
 class UniqueTreeDataset;
 
 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 grail::simgames::SimulatedGameThinkingUnit;
 };
}
}
}
 
#endif //GRAIL_OFFLINE_LEARNER_H