NICE_VisLearning/math/cluster/KMeansOnline.h

/** 
* @file KMeansOnline.h
* @brief online kmeans clustering
* @author Erik Rodner
* @date 02/13/2008

*/
#ifndef KMEANSONLINEINCLUDE
#define KMEANSONLINEINCLUDE

#ifdef NOVISUAL
#include <vislearning/nice_nonvis.h>
#else
#include <vislearning/nice.h>
#endif

#include "core/basics/Config.h"
#include "ClusterOnline.h"
#include "vislearning/cbaselib/Example.h"


namespace OBJREC {

/** online kmeans clustering */
class KMeansOnline : public ClusterOnline
{

    protected:
	size_t numClusters;
	double gamma;

    public:
  
	/** simple constructor */
	KMeansOnline( size_t numClusters, double gamma = 1.0 );
      
	/** simple destructor */
	virtual ~KMeansOnline();

	// refactor-nice.pl: check this substitution
	// old: int updateClusters ( const Vector & x );
	int updateClusters ( const NICE::Vector & x );
	void init ();
     
	/**
	 * Cluster algorithm using examples
	 * @param ex input examples
	 */
	void cluster( const Examples &ex);
	
	/**
	 * returns the distance to the centre of each cluster
	 * @param vin input vector
	 * @param dist distance
	 * @param size how much centres should considered
	 * @param hard use a hard assignment (all values = 1 or 0) or not
		 */
	void getDist(const NICE::Vector &vin, SparseVector &dist, int size = 1, bool hard = false);
};


} // namespace

#endif