Get cluster id of each location in data
| 30 | |
| 31 | // Get cluster id of each location in data |
| 32 | array clusterize(const array data, const array means) { |
| 33 | // Get manhattan distance |
| 34 | array dists = distance(data, means); |
| 35 | |
| 36 | // get the locations of minimum distance |
| 37 | array idx, val; |
| 38 | min(val, idx, dists, 1); |
| 39 | |
| 40 | // Return cluster IDs |
| 41 | return idx; |
| 42 | } |
| 43 | |
| 44 | array new_means(array data, array clusters, int k) { |
| 45 | int d = data.dims(2); |