TreeDataSample.hh

// Copyright QED Software 2023.
 
#ifndef GRAIL_TREE_DATA_SAMPLE_H
#define GRAIL_TREE_DATA_SAMPLE_H
 
#include <memory>
#include <vector>
#include "C45GameActionCounter.h"
 
namespace grail
{
namespace simgames
{
 class ISimulatedGameAction;
 
namespace learn
{
 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