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

Export Clusters.Nearest

Browse Source
This commit is contained in:
Christian Muehlhaeuser
2018-05-27 18:34:08 +02:00
parent 3cd22c49ed
commit f6a37b65ea
2 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cluster.go
View File

@@ -33,8 +33,8 @@ func (c Clusters) reset() {
} }
} }
// nearestCluster returns the index of the cluster nearest to point // Nearest returns the index of the cluster nearest to point
func (c Clusters) nearestCluster(point Point) int { func (c Clusters) Nearest(point Point) int {
var dist float64 var dist float64
var ci int var ci int

3
kmeans.go
View File

@@ -74,7 +74,7 @@ func (m Kmeans) Partition(dataset Points, k int) (Clusters, error) {
clusters.reset() clusters.reset()
for p, point := range dataset { for p, point := range dataset {
ci := clusters.nearestCluster(point) ci := clusters.Nearest(point)
clusters[ci].Points = append(clusters[ci].Points, point) clusters[ci].Points = append(clusters[ci].Points, point)
if points[p] != ci { if points[p] != ci {
points[p] = ci points[p] = ci
@@ -103,7 +103,6 @@ func (m Kmeans) Partition(dataset Points, k int) (Clusters, error) {
if m.debug { if m.debug {
draw(clusters, i) draw(clusters, i)
} }
if changes < int(float64(len(dataset))*m.deltaThreshold) { if changes < int(float64(len(dataset))*m.deltaThreshold) {
// fmt.Println("Aborting:", changes, int(float64(len(dataset))*m.TerminationThreshold)) // fmt.Println("Aborting:", changes, int(float64(len(dataset))*m.TerminationThreshold))
break break