2011-10-12 53 views
1

我想在我自己的文件上试试这个K-means聚类代码我该如何改变它,以便它不会创建随机信息,而是从我自己的数据源中读取它?kmeans matlab code feed自己的数据源

%% generate sample data 
K = 3; 
numObservarations = 100; 
dimensions = 3; 
data = rand([numObservarations dimensions]); 

%% cluster 
opts = statset('MaxIter', 500, 'Display', 'iter'); 
[clustIDX, clusters, interClustSum, Dist] = kmeans(data, K, 'options',opts, ... 
    'distance','sqEuclidean', 'EmptyAction','singleton', 'replicates',3); 

%% plot data+clusters 
figure, hold on 
scatter3(data(:,1),data(:,2),data(:,3), 50, clustIDX, 'filled') 
scatter3(clusters(:,1),clusters(:,2),clusters(:,3), 200, (1:K)', 'filled') 
hold off, xlabel('x'), ylabel('y'), zlabel('z') 

%% plot clusters quality 
figure 
[silh,h] = silhouette(data, clustIDX); 
avrgScore = mean(silh); 


%% Assign data to clusters 
% calculate distance (squared) of all instances to each cluster centroid 
D = zeros(numObservarations, K);  % init distances 
for k=1:K 
    %d = sum((x-y).^2).^0.5 
    D(:,k) = sum(((data - repmat(clusters(k,:),numObservarations,1)).^2), 2); 
end 

% find for all instances the cluster closet to it 
[minDists, clusterIndices] = min(D, [], 2); 

% compare it with what you expect it to be 
sum(clusterIndices == clustIDX) 
+0

您能否清楚地识别您发布的代码的哪部分是生成随机数据?如果是这样,那就是你应该存储数据集的变量。 – Pablo

+0

请告诉我们更多关于你的数据源的信息。它是在一个数据库中,在一个文件中(什么格式?),... –

+0

@Laurent',阅读OP的其他问题。 – Nzbuu

回答

4

创建中的随机数据线:

data = rand([numObservarations dimensions]); 

只是代码读取您的数据替换该行(可能使用MATLAB命令如textscan)到名为data变量。