UniqueTreeDataset.hh

#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 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