knnClassification.h

Go to the documentation of this file.

//
//  knnClassification.h
//  RapidLib
//
//  Created by mzed on 05/09/2016.
//  Copyright © 2016 Goldsmiths. All rights reserved.
//

#ifndef knnClassification_h
#define knnClassification_h

#include <vector>
#include "baseModel.h"

#ifndef EMSCRIPTEN
#include "../dependencies/json/json.h"
#endif

template<typename T>
class knnClassification : public baseModel<T> {
    
public:
    knnClassification(const int &num_inputs,
                      const std::vector<int> &which_inputs,
                      const std::vector<trainingExampleTemplate<T> > &trainingSet,
                      const int k);
    ~knnClassification();
    
    void addNeighbour(const int &classNum, const std::vector<T> &features);
    
    T run(const std::vector<T> &inputVector);
    
    void train(const std::vector<trainingExampleTemplate<T> > &trainingSet);
    
    void reset();
    
    int getNumInputs() const;
    
    std::vector<int> getWhichInputs() const;
    
    int getK() const;
    void setK(int newK);
    
#ifndef EMSCRIPTEN

    void getJSONDescription(Json::Value &currentModel);
#endif
    
private:
    int numInputs;
    std::vector<int> whichInputs;
    std::vector<trainingExampleTemplate<T>> neighbours;
    int desiredK; //K that user asked for might be limited but number of examples
    int currentK; //K minimum of desiredK or neighbours.size()
    inline void updateK();
    std::pair<int, T>* nearestNeighbours;
};

#endif