NaiveBayesClassifier/data_retriever.h at master · ramyan2/NaiveBayesClassifier · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#ifndef DATA_H
#define DATA_H

#include "image_reader.h"
#include <istream>
#include <ostream>
#include <iostream>
#include <string>
#include <vector>

using namespace std;


class DataRetriever {
    private:
            double k_value = 0.1;
    public:
    DataRetriever();
    DataRetriever(string training_images_file, string training_labels_file);

    bool RetrieveTrainingImages(string file);
    bool RetrieveTrainingLabels(string file);
    double CalculateLikelihoodForEachIndex(int row, int col, int class_number, int feature);
    double CalculatePriorsProbability(int class_number);
    std::vector<double> CreateVectorOfPriorsProbability();

        std::vector<ImagesReader> vector_of_images;
        std::vector<int> vector_of_labels;
        std::vector<double> vector_of_posterior_probabilities;
        double probability_model[28][28][10][2];
        static const int kImageLength = 28;
        static const int kNumberOfClasses = 10;

};
#endif