Update kMeans.md
This commit is contained in:
committed by
GitHub
parent
856703a41f
commit
31faa2ee2b
@ -18,8 +18,8 @@ const kMeans = (data, k = 1) => {
|
|||||||
Array.from({ length: k }, () => 0)
|
Array.from({ length: k }, () => 0)
|
||||||
);
|
);
|
||||||
const classes = Array.from({ length: data.length }, () => -1);
|
const classes = Array.from({ length: data.length }, () => -1);
|
||||||
|
|
||||||
let itr = true;
|
let itr = true;
|
||||||
|
|
||||||
while (itr) {
|
while (itr) {
|
||||||
itr = false;
|
itr = false;
|
||||||
|
|
||||||
@ -29,7 +29,6 @@ const kMeans = (data, k = 1) => {
|
|||||||
...Object.keys(data[0]).map(key => data[d][key] - centroids[c][key])
|
...Object.keys(data[0]).map(key => data[d][key] - centroids[c][key])
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
const m = distances[d].indexOf(Math.min(...distances[d]));
|
const m = distances[d].indexOf(Math.min(...distances[d]));
|
||||||
if (classes[d] !== m) itr = true;
|
if (classes[d] !== m) itr = true;
|
||||||
classes[d] = m;
|
classes[d] = m;
|
||||||
@ -44,7 +43,6 @@ const kMeans = (data, k = 1) => {
|
|||||||
}
|
}
|
||||||
return acc;
|
return acc;
|
||||||
}, 0);
|
}, 0);
|
||||||
|
|
||||||
for (let i in data[0]) {
|
for (let i in data[0]) {
|
||||||
centroids[c][i] = parseFloat(Number(centroids[c][i] / size).toFixed(2));
|
centroids[c][i] = parseFloat(Number(centroids[c][i] / size).toFixed(2));
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user