2011-10-12 46 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變量。