66 lines
1.4 KiB
JavaScript
66 lines
1.4 KiB
JavaScript
const skmeans = require("../main.js");
|
|
const data1 = require("./data/well-separated.js");
|
|
const data2 = require("./data/random_pts.js");
|
|
const MAX = 1000;
|
|
|
|
function testCentroids(ks) {
|
|
ks.forEach(k=>{
|
|
k.forEach(v=>{
|
|
if(v==0) {
|
|
console.error(`Centroid on 0 ${k}`);
|
|
process.exit(1);
|
|
}
|
|
})
|
|
})
|
|
}
|
|
|
|
console.log("Simple data (k=3)");
|
|
var its = 0;
|
|
for(var i=0;i<MAX;i++) {
|
|
var res = skmeans(data1,3);
|
|
testCentroids(res.centroids);
|
|
its += res.it;
|
|
}
|
|
console.log("skmeans default\t\t=>",its/MAX);
|
|
|
|
var its = 0;
|
|
for(var i=0;i<MAX;i++) {
|
|
var res = skmeans(data1,3,"kmrand");
|
|
testCentroids(res.centroids);
|
|
its += res.it;
|
|
}
|
|
console.log("skmeans kmrand\t\t=>",its/MAX);
|
|
|
|
var its = 0;
|
|
for(var i=0;i<MAX;i++) {
|
|
var res = skmeans(data1,3,"kmpp");
|
|
testCentroids(res.centroids);
|
|
its += res.it;
|
|
}
|
|
console.log("skmeans kmpp\t\t=>",its/MAX);
|
|
|
|
console.log("Complex data (k=16)");
|
|
var its = 0;
|
|
for(var i=0;i<MAX;i++) {
|
|
var res = skmeans(data2,16);
|
|
testCentroids(res.centroids);
|
|
its += res.it;
|
|
}
|
|
console.log("skmeans default\t\t=>",its/MAX);
|
|
|
|
var its = 0;
|
|
for(var i=0;i<MAX;i++) {
|
|
var res = skmeans(data2,16,"kmrand");
|
|
testCentroids(res.centroids);
|
|
its += res.it;
|
|
}
|
|
console.log("skmeans kmrand\t\t=>",its/MAX);
|
|
|
|
var its = 0;
|
|
for(var i=0;i<MAX;i++) {
|
|
var res = skmeans(data2,16,"kmpp");
|
|
testCentroids(res.centroids);
|
|
its += res.it;
|
|
}
|
|
console.log("skmeans kmpp\t\t=>",its/MAX);
|