Dataset.hh

// Copyright QED Software 2023.
 
#ifndef GRAIL_DATASET_H
#define GRAIL_DATASET_H
 
#include <functional>
#include <initializer_list>
#include <iostream>
#include <memory>
#include <vector>
#include "DatasetSample.hh"
#include "../DecisionTree/DecisionNode.h"
#include "../DecisionTree/DTConsiderationType.hh"
 
namespace grail
{
namespace simgames
{
 class ISimulatedGameAction;
 
namespace learn
{
 class Dataset
 {
 public:
 size_t DecisionVariablesCount() const;
 
 const std::vector<dt::DTConsiderationType>& GetConsiderationTypes() const;
 
 dt::DTConsiderationType GetConsiderationType(int columnIndex) const;
 
 Dataset(std::initializer_list<dt::DTConsiderationType> considerationTypes);
 
 Dataset(const std::vector<dt::DTConsiderationType>& considerationTypes);
 
 void AddSample(std::unique_ptr<ISimulatedGameAction> decision, std::initializer_list<float> data);
 
 void AddSample(std::unique_ptr<ISimulatedGameAction> decision, const std::vector<float>& data);
 
 void AddSample(std::unique_ptr<DatasetSample> sample);
 
 double ValidateBinary(dt::DecisionNode<ISimulatedGameAction>& decisionTreeNodeRoot);
 
 std::vector<std::unique_ptr<DatasetSample>> Samples;
 
 bool MoveFromOther(Dataset& sourceDataset);
 
 bool CopyFromOther(Dataset& sourceDataset);
 
 private:
 bool CheckDataCompatibility(const Dataset& other) const;
 
 double CalculateDecisionEntropy() const;
 
 std::unique_ptr<Dataset> SplitByFilter(std::function<bool(std::vector<float>&)> filter) const;
 std::vector<std::unique_ptr<Dataset>> SplitContinuous(size_t column, float splitValue) const;
 std::vector<std::pair<std::unique_ptr<Dataset>, float>> SplitNominal(int column) const;
 
 std::vector<dt::DTConsiderationType> considerationTypes;
 
 friend class C45Algorithm;
 };
}
}
}
 
#endif //GRAIL_DATASET_H