X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/841662dc25b40dce151a0bf7a024e55682028f18..33b793fe326a2706173b38d2fc4b454982fbf900:/src/k_means.py

diff --git a/src/k_means.py b/src/k_means.py
index f429824..119f576 100644
--- a/src/k_means.py
+++ b/src/k_means.py
@@ -4,11 +4,11 @@ import random
 
 def cluster(k, d, data, i_centers=None):
     """Find *k* clusters on *d* dimensional *data*."""
-    borders = [(min(p[0][i] for p in data), max(p[0][i] for p in data))
-               for i in range(d) ]
     if i_centers:
         old_centers = i_centers
     else:
+        borders = [(min(p[0][i] for p in data), max(p[0][i] for p in data))
+               for i in range(d)]
         old_centers = [[(h - l) * random.random() + l for (l, h) in borders]
                for _ in range(k)]
     clusters, centers = next_step(old_centers, data)