k-means-clusterer.h

Go to the documentation of this file.

// Copyright 2007, Massachusetts Institute of Technology.
// The use of this code is permitted for research only. There is
// absolutely no warranty for this software.
//
// Author: John Lee (jjl@mit.edu)
//

#ifndef CLUSTERING_K_MEANS_CLUSTERER_H
#define CLUSTERING_K_MEANS_CLUSTERER_H

#include <vector>
#include "clustering/clusterer.h"
#include "point_set/point-ref.h"
#include "util/distance-computer.h"

using namespace std;

namespace libpmk {

class KMeansClusterer : public Clusterer {
public:
  KMeansClusterer(int num_clusters, int max_iters,
                  const DistanceComputer& distance_computer);

protected:

  virtual void DoClustering(const vector<PointRef>& data);

private:
  int num_clusters_;
  int max_iterations_;
  const DistanceComputer& distance_computer_;

  void ComputeMembership(const vector<PointRef>& data);

  // Returns true if this call to ComputeMeans() has changed
  // cluster_centers_.
  bool ComputeMeans(const vector<PointRef>& data);
};
}  // namespace libpmk

#endif  // CLUSTERING_K_MEANS_CLUSTERER_H

---