高斯混合模型直观实现

vs2010环境下编写的VC++代码，EM算法直观实现GMM。 效果参照：http://blog.csdn.net/hzq20081121107/article/details/19505619

#include “StdAfx.h“
#include 
#include “ClassGMM.h“

#define PI acos（-1.0）


ClassGMM::ClassGMM（void）
{
	numSample=0;
	numClass=0;
	pGM=NULL;
	pSampleNode=NULL;
}


ClassGMM::~ClassGMM（void）
{
	if（pGM!=NULL）
	{
	    delete []pGM;
		pGM=NULL;
	}
	if（pSampleNode!=NULL）
	{
	    delete []pSampleNode;
		pSampleNode=NULL;
	}
}


void ClassGMM::clear（void）
{
	numSample=0;
	numClass=0;
	if（pGM!=NULL）
	{
		delete []pGM;
		pGM=NULL;
	}
	if（pSampleNode!=NULL）
	{
		delete []pSampleNode;
		pSampleNode=NULL;
	}
}

//EM迭代 nTimes
int ClassGMM::EM（int nTimes）
{
	int ijk;
	//开辟空间
	float **ppR;
	ppR=new float* [numSample];
	for（i=0;i	{
	    ppR[i]=new float[numClass];
	}


	while（--nTimes>=0）
	{
		//计算r（ik），用ppR[ik]表示
		for（i=0;i		{
			float *piKN;
			piKN=new float[numClass];//
			float sum_piKN=0;
			for（k=0;k			{
				piKN[k]=pGM[k].piK*getGaussRatio（pGM[k].upGM[k].sigmapSampleNode[i].x）;
				sum_piKN+=piKN[k];
			}
			for（k=0;k			{
				ppR[i][k]=piKN[k]/sum_piKN;
			}
			delete []piKN;
		}
		//利用r（ik）求高斯参数，Nk，piK，u，sigma，
		//Nk
		for（k=0;k		{
			pGM[k].Nk=0;
			for（i=0;i			{
				pGM[k].Nk += ppR[i][k];
			}
		}
		//piK
		float sumNK=0;
		for（k=0;k		{
			sumNK+=pGM[k].Nk;
		}
		for（k=0;k		{
			pGM[k].piK=pGM[k].Nk/sumNK;
		}
		//u
		for（k=0;k		{
			pGM[k].u=0;
			for（i=0;i			{
				pGM[k].u+=pSampleNode[i].x*ppR[i][k];
			}
			pGM[k].u/=pGM[k].Nk;
		}
		//sigma
		for（k=0;k		{
			float sum_temp=0;
			for（i=0;i			{
				sum_temp+=ppR[i][k]*（pSampleNode[i].x-pGM[k].u）*（pSampleNode[i].x-pGM[k].u）;
			}
			sum_temp/=pGM[k].Nk;
			pGM[k].sigma=sqrt（sum_temp）;
		}
	}
	//销毁空间
	for（i=0;i	{
		delete []ppR[i];
	}
	delete []ppR;

	return 0;
}

//求高斯概率密度
float ClassGMM::getGaussRatio（float u  float sigma  float x）
{
	double fenMu=sqrt（2*PI）*sigma;
	double zhiShu=-（x-u）*（x-u）/（2*sigma*sigma）;
	double fenZi=exp（zhiShu）;
	return fenZi/fenMu;
}

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----

     文件       2233  2014-02-14 15:53  GMM_test2\GMM_test\ClassGMM.cpp

     文件        391  2014-02-14 16:04  GMM_test2\GMM_test\ClassGMM.h

     文件     106512  2014-02-19 21:35  GMM_test2\GMM_test\GMM_test.aps

     文件       2019  2014-01-20 14:55  GMM_test2\GMM_test\GMM_test.cpp

     文件        454  2014-01-20 14:55  GMM_test2\GMM_test\GMM_test.h

     文件      10542  2014-02-19 21:35  GMM_test2\GMM_test\GMM_test.rc

     文件       6012  2014-02-14 11:10  GMM_test2\GMM_test\GMM_test.vcxproj

     文件       2177  2014-02-14 11:10  GMM_test2\GMM_test\GMM_test.vcxproj.filters

     文件        143  2014-01-20 14:55  GMM_test2\GMM_test\GMM_test.vcxproj.user

     文件       6800  2014-02-19 21:34  GMM_test2\GMM_test\GMM_testDlg.cpp

     文件       1098  2014-02-19 21:25  GMM_test2\GMM_test\GMM_testDlg.h

     文件       3160  2014-01-20 14:55  GMM_test2\GMM_test\ReadMe.txt

     文件      67777  2009-08-31 02:31  GMM_test2\GMM_test\res\GMM_test.ico

     文件        672  2014-01-20 14:55  GMM_test2\GMM_test\res\GMM_test.rc2

     文件       1822  2014-01-20 16:50  GMM_test2\GMM_test\resource.h

     文件        141  2014-01-20 14:55  GMM_test2\GMM_test\stdafx.cpp

     文件       1632  2014-01-20 14:55  GMM_test2\GMM_test\stdafx.h

     文件        234  2014-01-20 14:55  GMM_test2\GMM_test\targetver.h

     文件        891  2014-01-20 14:55  GMM_test2\GMM_test.sln

    ..A..H.     29184  2014-02-19 21:44  GMM_test2\GMM_test.suo

     目录          0  2014-02-13 17:26  GMM_test2\GMM_test\res

     目录          0  2014-02-26 21:02  GMM_test2\GMM_test

     目录          0  2014-02-26 21:02  GMM_test2

----------- ---------  ---------- -----  ----

               243894                    23