classification/KMeanCluster.h

Go to the documentation of this file.

#ifndef _KMEANCLUSTER_H
#define _KMEANCLUSTER_H
#include <list>
#include "../util/Vector.h"


class KMeanCluster 
{
    public:
        typedef std::list<Vector3d> VectorList;

        KMeanCluster(int nCentres);

        void train(VectorList &v);

        int clusterize(Vector3d &v);

        VectorList::iterator begin() { return m_centres.begin(); }

        VectorList::iterator end() { return m_centres.end(); }

        const size_t getClusterCount() const { return m_centres.size(); }

        void setCentres(VectorList &centres) { m_centres = centres; }

    private:
        bool isSameSet(VectorList &a, VectorList &b);
        VectorList m_centres;
};

#endif

---