KNN.h

Go to the documentation of this file.

 
#ifndef __KNN_H__
#define __KNN_H__
 
#include <dignea/core/Solution.h>
#include <dignea/distances/Distance.h>
#include <dignea/searches/NoveltySearch.h>
 
#include <functional>
#include <iostream>
#include <string>
#include <utility>
#include <vector>
 
using namespace std;
 
using neighborT = pair<float, int>;
 
using neighVector = vector<neighborT>;
using neighMatrix = vector<neighVector>;
 
using Descriptor = vector<float>;
class KNN {
   public:
    KNN() = default;
 
    virtual ~KNN() = default;
 
    neighVector run(const Descriptor &individual,
                    const vector<Descriptor> &neighbours,
                    Distance<float> *metric, const int &k = 3);
 
    neighMatrix run(const vector<Descriptor> &population,
                    Distance<float> *metric, const int &k = 3);
};
 
vector<float> sparseness(const vector<Descriptor> &population,
                         Distance<float> *metric, const int k);
float sparseness(const Descriptor &solution,
                 const vector<Descriptor> &population, Distance<float> *metric,
                 const int k);
 
#endif  // __KNN_H__