neingeist/coursera-ml-007-exercises
Archived
1
0
Fork 0
Code Releases Activity

Compute centroid means (vectorized)

Browse source
This commit is contained in:
neingeist 2014-11-17 23:13:57 +01:00
parent 39b09f144a
commit 6c51a29ca2
1 changed files with 3 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

14
ex7/computeCentroids.m
View file

@ -27,17 +27,9 @@ centroids = zeros(K, n);
%
for k = 1:K
count = 0;
% XXX vectorize
for i = 1:m
if idx(i) == k
centroids(k, :) += X(i, :);
count += 1;
end
end
if count > 0
centroids(k, :) /= count;
X_in_k = X(idx == k, :);
if size(X_in_k, 1) > 0
centroids(k, :) = sum(X_in_k)/size(X_in_k, 1);
end
end