kmeans.h

简单明了的HMM-GMM-KEAMS调用接口

/***************************************************************************

Module Name:

	KMeans

History: 

	2003/10/16	Fei Wang

***************************************************************************/


#pragma once
#include <fstream>


class KMeans
{
public:
	enum InitMode
	{
		InitRandom,
		InitManual,
		InitUniform,
	};

	KMeans(int dimNum = 1, int clusterNum = 1);
	~KMeans();

	void SetMean(int i, const double* u){ memcpy(m_means[i], u, sizeof(double) * m_dimNum); }
	void SetInitMode(int i)				{ m_initMode = i; }
	void SetMaxIterNum(int i)			{ m_maxIterNum = i; }
	void SetEndError(double f)			{ m_endError = f; }

	double* GetMean(int i)	{ return m_means[i]; }
	int GetInitMode()		{ return m_initMode; }
	int GetMaxIterNum()		{ return m_maxIterNum; }
	double GetEndError()	{ return m_endError; }

	
	/*	SampleFile: <size><dim><data>...
		LabelFile:	<size><label>...
	*/
	void Cluster(const char* sampleFileName, const char* labelFileName);

	void Cluster(double *data, int N, int *Label);

	void Init(double *data, int N);

	friend std::ostream& operator<<(std::ostream& out, KMeans& kmeans);

	double GetLabel(const double* x, int* label);

private:
	int m_dimNum;
	int m_clusterNum;

	double** m_means;

	int m_initMode;
	int m_maxIterNum;		// The stopping criterion regarding the number of iterations
	double m_endError;		// The stopping criterion regarding the error

	void Init(std::ifstream& sampleFile);
//	double GetLabel(const double* x, int* label);
	double CalcDistance(const double* x, const double* u, int dimNum);
};