UniqueTreeDataset.hh

// Copyright QED Software 2023.
 
#ifndef GRAIL_UNIQUE_TREE_DATASET_H
#define GRAIL_UNIQUE_TREE_DATASET_H
 
#include <initializer_list>
#include <memory>
#include <unordered_set>
#include "../ISimulatedGameAction.hh"
#include "../DecisionTree/DTConsiderationType.hh"
 
namespace grail
{
namespace simgames
{
 class ISimulatedGameAction;
 
namespace learn
{
 class TreeDataSample;
 class Dataset;
 
 class UniqueTreeDataset
 {
 public:
 virtual ~UniqueTreeDataset();
 
 void Add(std::unique_ptr<TreeDataSample> sample);
 
 void Add(std::vector<float>& data, const ISimulatedGameAction* action);
 
 std::unique_ptr<Dataset> ConvertToDataset(
 const std::initializer_list<dt::DTConsiderationType>& considerationTypes);
 
 std::unique_ptr<Dataset> ConvertToDataset(const std::vector<dt::DTConsiderationType>& considerationTypes);
 
 std::unique_ptr<Dataset> ConvertToDataset(dt::DTConsiderationType considerationType, int count);
 private:
 std::unordered_set<std::unique_ptr<TreeDataSample>, HashForOfflineLearning, EqualsForOfflineLearning>
 samples;
 };
}
}
}
 
#endif //GRAIL_UNIQUE_TREE_DATASET_H