centroid test

solzimer · Jul 20, 2017 · f618769 · f618769
1 parent e5c7755
commit f618769
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/test/stats.js b/test/stats.js
@@ -3,24 +3,38 @@ 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);
@@ -29,20 +43,23 @@ 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);