gmmtraindemo2dcovtype02.m

来自「一个关于数据聚类和模式识别的程序,在生物化学,化学中因该都可以用到.希望对大家有」· M 代码 · 共 26 行

% Animation of GMM training with covType=2 (diagonal cov. matrix) for 2D data

% Data collection
DS = dcData(2);
data=DS.input;

% GMM training
covType=2;
gaussianNum=4;
gmmTrainParam=gmmTrainParamSet;
gmmTrainParam.useKmeans=0;
gmmTrainParam.dispOpt=1;
gmmTrainParam.maxIteration=100;
[gmmParam, lp] = gmmTrain(data, [gaussianNum, covType], gmmTrainParam);

% Surface and contour plots
pointNum = 40;
x = linspace(min(data(1,:)), max(data(1,:)), pointNum);
y = linspace(min(data(2,:)), max(data(2,:)), pointNum);
[xx, yy] = meshgrid(x, y);
data = [xx(:) yy(:)]';
z = gmmEval(data, gmmParam);
zz = reshape(z, pointNum, pointNum);
figure; mesh(xx, yy, zz); axis tight; box on; rotate3d on
figure; contour(xx, yy, zz, 30); axis image