9 #ifndef KNNCLASSIFICATION_H
10 #define KNNCLASSIFICATION_H
16 #include "../dependencies/json/json.h"
31 const std::vector<size_t> &which_inputs,
41 void addNeighbour(
const int classNum,
const std::vector<T>& features);
47 T
run(
const std::vector<T>& inputVector)
override;
64 void reset()
override;
93 std::vector<size_t> whichInputs {};
94 std::size_t whichOutput {};
95 std::vector<trainingExampleTemplate<T>> neighbours {};
98 inline void updateK();
Definition: baseModel.h:42
Definition: knnClassification.h:22
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
size_t getNumInputs() const override
Definition: knnClassification.cpp:40
void addNeighbour(const int classNum, const std::vector< T > &features)
Definition: knnClassification.cpp:68
T run(const std::vector< T > &inputVector) override
Definition: knnClassification.cpp:94
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