WIP: added metric support to lloyd

Author: PyDataBlog

src/lloyd.jl

@@ -118,7 +118,7 @@ function chunk_update_centroids(::Lloyd, containers, centroids, X, weights, metr
         min_dist = distance(metric, X, centroids, i, 1)
         label = 1
         for j in 2:size(centroids, 2)
-            dist = distance(metric, X, centroids, i, 1)
+            dist = distance(metric, X, centroids, i, j)
             label = dist < min_dist ? j : label
             min_dist = dist < min_dist ? dist : min_dist
         end

src/yinyang.jl

@@ -222,7 +222,7 @@ function calculate_centroids_movement(alg::Yinyang, containers, centroids)
     end
 end
 
-function chunk_update_centroids(alg, containers, centroids, X, weights, r, idx)
+function chunk_update_centroids(alg::Yinyang, containers, centroids, X, weights, r, idx)
     # unpack containers for easier manipulations
     centroids_new = containers.centroids_new[idx]
     centroids_cnt = containers.centroids_cnt[idx]