UniqueTreeDataset.hh

#ifndef GRAIL_UNIQUE_TREE_DATASET_H
#define GRAIL_UNIQUE_TREE_DATASET_H
 
#include "../ISimulatedGameAction.hh"
#include "../DT/DTConsiderationType.hh"
#include <unordered_set>
#include <initializer_list>
#include <memory>
 
namespace grail
{
 namespace simulation
 {
 class TreeDataSample;
 class ISimulatedGameAction;
 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<DTConsiderationType>& considerationTypes);
 
 std::unique_ptr<Dataset> ConvertToDataset(const std::vector<DTConsiderationType>& considerationTypes);
 
 std::unique_ptr<Dataset> ConvertToDataset(DTConsiderationType considerationType, int count);
 private:
 std::unordered_set<std::unique_ptr<TreeDataSample>, HashForOfflineLearning, EqualsForOfflineLearning> samples;
 };
 }
}
 
#endif //GRAIL_UNIQUE_TREE_DATASET_H