KMeans.h

#ifndef STAT_KMEANS_H
#define STAT_KMEANS_H

#include <KrisLibrary/math/vector.h>
#include <vector>

namespace Statistics {
  using namespace Math;

class KMeans
{
 public:
  KMeans(const std::vector<Vector>& data);
  KMeans(const std::vector<Vector>& data,int k);
  virtual ~KMeans() {}
  int GetK() const { return (int)centers.size(); }
  void SetK(int k);
  void RandomInitialCenters();
  void ClearLabels();
  void Iterate(int& maxIters);

  //Individual steps of the iteration
  bool CalcLabelsFromCenters();
  void CalcCentersFromLabels();

  Real AverageDistance(int c);
  void AverageDistance(std::vector<Real>& dist);

  virtual Real Distance(const Vector& a,const Vector& b) 
  { return a.distance(b); }

  const std::vector<Vector>& data;
  const std::vector<Real>* weights;
  std::vector<int> labels;
  std::vector<Vector> centers;
};

}

#endif