1 #ifndef knnClassification_h 2 #define knnClassification_h 22 const std::vector<int> &which_inputs,
23 const std::vector<trainingExample> &trainingSet,
31 void addNeighbour(
const int &classNum,
const std::vector<double> &features);
37 double process(
const std::vector<double> &inputVector);
44 void train(
const std::vector<trainingExample> &trainingSet);
55 std::vector<int> whichInputs;
56 std::vector<trainingExample> neighbours;
58 std::pair<int, double>* nearestNeighbours;
~knnClassification()
Definition: knnClassification.cpp:19
Definition: knnClassification.h:12
knnClassification(const int &num_inputs, const std::vector< int > &which_inputs, const std::vector< trainingExample > &trainingSet, const int &k)
Definition: knnClassification.cpp:10
int getNumInputs()
Definition: knnClassification.cpp:23
void addNeighbour(const int &classNum, const std::vector< double > &features)
Definition: knnClassification.cpp:31
Definition: baseModel.h:12
void train(const std::vector< trainingExample > &trainingSet)
Definition: knnClassification.cpp:38
void getJSONDescription(Json::Value ¤tModel)
Definition: knnClassification.cpp:111
std::vector< int > getWhichInputs()
Definition: knnClassification.cpp:27
double process(const std::vector< double > &inputVector)
Definition: knnClassification.cpp:43