package/kmeans
1
0
Fork 0
mirror of https://github.com/muesli/kmeans.git synced 2025-09-27 03:56:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added Cluster.removePoint(point) for convenience

Browse Source
This commit is contained in:
Christian Muehlhaeuser
2018-05-29 03:54:58 +02:00
parent ff5e403929
commit c49d1033bc
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
cluster.go
View File

@@ -26,6 +26,18 @@ func (c Clusters) Nearest(point Point) int {
return ci return ci
} }
// removePoint removes a point's assignment to this cluster
func (c *Cluster) removePoint(p Point) {
var pp Points
for _, v := range c.Points {
if !v.Equal(p) {
pp = append(pp, p)
}
}
c.Points = pp
}
// recenter recenters a cluster // recenter recenters a cluster
func (c *Cluster) recenter() { func (c *Cluster) recenter() {
center, err := c.Points.Mean() center, err := c.Points.Mean()