multi.js 375 B

123456789101112131415
  1. const skmeans = require("../main.js");
  2. var data = [];
  3. for(let i=0;i<2000;i++) {
  4. var r = Math.random();
  5. if(r>=0 && r<0.33) data.push([r,r*2]);
  6. else if(r>=0.33 && r<0.66) data.push([r+4,2*(r+4)]);
  7. else data.push([r+9,2*(r+9)]);
  8. }
  9. var res = skmeans(data,3,"kmrand");
  10. console.log(res.it,res.centroids);
  11. var res = skmeans(data,3,"kmpp");
  12. console.log(res.it,res.centroids);