10 #ifndef KNNCLASSIFICATION_H
11 #define KNNCLASSIFICATION_H
17 #include "../dependencies/json/json.h"
32 const std::vector<size_t> &which_inputs,
42 void addNeighbour(
const int classNum,
const std::vector<T>& features);
48 T
run(
const std::vector<T>& inputVector)
const override;
65 void reset()
override;
94 std::vector<size_t> whichInputs {};
95 std::size_t whichOutput {};
96 std::vector<trainingExampleTemplate<T>> neighbours {};
99 inline void updateK();
Definition: baseModel.h:42
Definition: knnClassification.h:23
std::vector< size_t > getWhichInputs() const override
Definition: knnClassification.cpp:46
void getJSONDescription(Json::Value ¤tModel) override
Definition: knnClassification.cpp:179
void reset() override
Definition: knnClassification.cpp:35
int getK() const
Definition: knnClassification.cpp:51
~knnClassification()
Definition: knnClassification.cpp:33
void setK(int newK)
Definition: knnClassification.cpp:61
T run(const std::vector< T > &inputVector) const override
Definition: knnClassification.cpp:94
size_t getNumInputs() const override
Definition: knnClassification.cpp:40
void addNeighbour(const int classNum, const std::vector< T > &features)
Definition: knnClassification.cpp:68
void train(const std::vector< trainingExampleTemplate< T > > &trainingSet) override
knnClassification(const int &num_inputs, const std::vector< size_t > &which_inputs, const std::vector< trainingExampleTemplate< T > > &trainingSet, const int k)
Definition: knnClassification.cpp:20
Definition: trainingExample.h:19