k-means

1function kMeans(pts: Array<[x: number, y: number]>, k: number, iters: number) {
2  let centroids = pts.slice(0, k) // первые k точек как старт
3 
4  for (let it = 0; it < iters; it++) {
5    // 1. назначение: каждая точка к ближайшему центроиду
6    const assign = pts.map((p) =>
7      argmin(centroids, (c) => dist2(p, c)),
8    )
9 
10    // 2. пересчёт: центроид = среднее своих точек
11    centroids = range(k).map((c) => {
12      const mine = pts.filter((_, i) => assign[i] === c)
13      return avg(mine)
14    })
15  }
16 
17  return { centroids }
18}