TreeDataSample.hh

#ifndef GRAIL_TREE_DATA_SAMPLE_H
#define GRAIL_TREE_DATA_SAMPLE_H
 
#include <memory>
#include <vector>
#include "C45GameActionCounter.h"
 
namespace grail
{
    namespace simulation
    {
        class ISimulatedGameAction;
        struct DatasetSample;
 
        class TreeDataSample
        {
        public:
            TreeDataSample(std::vector<float>& initData);
            TreeDataSample(std::vector<float>& initData, const ISimulatedGameAction* decision);
 
            virtual ~TreeDataSample();
 
            void AddDecision(const ISimulatedGameAction* decision);
            std::unique_ptr<DatasetSample> Reduce();
            const ISimulatedGameAction* GetFirstDecision() const;
 
            size_t HashForLearning() const;
            bool EqualsForLearning(const TreeDataSample& other) const;
 
            std::vector<float> data;
 
        private:
            C45GameActionCounter counter;
        };
    }
}
#endif //GRAIL_TREE_DATA_SAMPLE_H