NaiveBayesClassifier/classifier.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
#ifndef CLASSIFIER_H
#define CLASSIFIER_H

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

using namespace std;


class Classifier {
    public:
    double probability_model[28][28][10][2];
    std::vector<double> priors_vector;
    static const int kImageLength = 28;
    static const int kNumberOfClasses = 10;
    Classifier();
    Classifier(string images_data_file, string labels_data_file);
    void SetToZero();
    void SetLikelihoodToModel(DataRetriever training_model);
    void CreateProbabilityModel(DataRetriever training_model);
    void CreatePriorsModel(DataRetriever training_model);
    int CalculatePosteriorProbabilities(ImagesReader image);
    void LoadModelFromFile(string probability_file);
    void LoadPriorsModelFromFile(string priors_file);
    void SaveModelToFile(string probability_file);
    void SavePriorsModelToFile(string priors_file);
};
#endif