knnClassification.h

Go to the documentation of this file.

 
#pragma once
 
#include "baseModel.h"
 
#ifndef EMSCRIPTEN
#include "../dependencies/json/json.h"
#endif
 
#include <optional>
#include <vector>
 
template<typename T>
class knnClassification final : public baseModel<T>
{
  public:
  knnClassification(const std::optional<size_t> num_inputs,
                    const std::vector<size_t> &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) const override;
 
  void train(const std::vector<trainingExampleTemplate<T> >& trainingSet) override;
 
  void train(const std::vector<trainingExampleTemplate<T>>& trainingSet, const std::size_t whichOutput) override;
 
  void reset() override;
 
  size_t getNumInputs() const override;
 
  std::vector<size_t> getWhichInputs() const override;
 
  size_t getK() const;
 
  void setK(int newK);
 
#ifndef EMSCRIPTEN
  void getJSONDescription(Json::Value& currentModel) const override;
#endif
 
  private:
  size_t numInputs {};
  std::vector<size_t> whichInputs {};
  std::size_t whichOutput {};
  std::vector<trainingExampleTemplate<T>> neighbours {};
  size_t desiredK {}; //K that user asked for might be limited but number of examples
  size_t currentK {}; //K minimum of desiredK or neighbours.size()
  inline void updateK();
};