Compute centroid means (vectorized)
This commit is contained in:
		
							parent
							
								
									39b09f144a
								
							
						
					
					
						commit
						6c51a29ca2
					
				
					 1 changed files with 3 additions and 11 deletions
				
			
		| 
						 | 
					@ -27,17 +27,9 @@ centroids = zeros(K, n);
 | 
				
			||||||
%
 | 
					%
 | 
				
			||||||
 | 
					
 | 
				
			||||||
for k = 1:K
 | 
					for k = 1:K
 | 
				
			||||||
  count = 0;
 | 
					  X_in_k = X(idx == k, :);
 | 
				
			||||||
 | 
					  if size(X_in_k, 1) > 0
 | 
				
			||||||
  % XXX vectorize
 | 
					    centroids(k, :) = sum(X_in_k)/size(X_in_k, 1);
 | 
				
			||||||
  for i = 1:m
 | 
					 | 
				
			||||||
    if idx(i) == k
 | 
					 | 
				
			||||||
      centroids(k, :) += X(i, :);
 | 
					 | 
				
			||||||
      count += 1;
 | 
					 | 
				
			||||||
    end
 | 
					 | 
				
			||||||
  end
 | 
					 | 
				
			||||||
  if count > 0
 | 
					 | 
				
			||||||
    centroids(k, :) /= count;
 | 
					 | 
				
			||||||
  end
 | 
					  end
 | 
				
			||||||
end
 | 
					end
 | 
				
			||||||
 | 
					
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Reference in a new issue