TreeDataSample.hh

#ifndef GRAIL_TREE_DATA_SAMPLE_H
#define GRAIL_TREE_DATA_SAMPLE_H
 
#include "C45GameActionCounter.h"
#include <vector>
#include <memory>
 
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