kmeans

README

KMeans-Go

Implementation of KMeans Clustering in Golang.

Documentation

Index

• func LabelCount(labels []int, k int) []int
• type KMeans
  • func (kmeans *KMeans) AddPoint(point Point)
  • func (kmeans *KMeans) AddPointList(points []Point)
  • func (kmeans *KMeans) CentroidsToString() string
  • func (kmeans *KMeans) ComputeLabels() []int
  • func (kmeans *KMeans) ComputeSSE() float64
  • func (kmeans *KMeans) Fit(pointList []Point) bool
  • func (kmeans *KMeans) GetPointIdxOfCentroid(centroidIdx int) []int
  • func (kmeans *KMeans) InitCentroids() bool
  • func (kmeans *KMeans) LabelCount() []int
  • func (kmeans *KMeans) LabelsToString() string
  • func (kmeans *KMeans) PointsToString() string
• type Point
  • func ComputeCentroids(pointList []Point, distIndex []int, k int) []Point
  • func (point Point) Norm() float64
  • func (point Point) PointDist(p2 Point) float64
  • func (point Point) PointEqual(p2 Point) bool
  • func (point Point) Print()
  • func (point Point) Subtract(p2 Point) Point

Functions

func LabelCount

func LabelCount(labels []int, k int) []int

LabelCount - Returns the number of points per centroid

Types

type KMeans

type KMeans struct {
	// contains filtered or unexported fields
}

KMeans - Struct for KMeans Algorithm

func (*KMeans) AddPoint

func (kmeans *KMeans) AddPoint(point Point)

AddPoint - Add a new Point to the list and recalculate centroids

func (*KMeans) AddPointList

func (kmeans *KMeans) AddPointList(points []Point)

AddPointList - Add a list of Points to the list and recalculate centroids

func (*KMeans) CentroidsToString

func (kmeans *KMeans) CentroidsToString() string

CentroidsToString - returns the string of centroids

func (*KMeans) ComputeLabels

func (kmeans *KMeans) ComputeLabels() []int

ComputeLabels - Computes labels of the points

func (*KMeans) ComputeSSE

func (kmeans *KMeans) ComputeSSE() float64

ComputeSSE - computes Sum of Squared Erros

func (*KMeans) Fit

func (kmeans *KMeans) Fit(pointList []Point) bool

Fit - Fits the points into k centroids

func (*KMeans) GetPointIdxOfCentroid

func (kmeans *KMeans) GetPointIdxOfCentroid(centroidIdx int) []int

GetPointIdxOfCentroid - return indexes of points of a certain centroid

func (*KMeans) InitCentroids

func (kmeans *KMeans) InitCentroids() bool

InitCentroids - Inits the Centroids for random

func (*KMeans) LabelCount

func (kmeans *KMeans) LabelCount() []int

LabelCount - Returns the number of points per centroid

func (*KMeans) LabelsToString

func (kmeans *KMeans) LabelsToString() string

LabelsToString - Returns the string of Labels

func (*KMeans) PointsToString

func (kmeans *KMeans) PointsToString() string

PointsToString - Returns the String of the Points

type Point

type Point []float64

Point - Slice of Float64 Values

func ComputeCentroids

func ComputeCentroids(pointList []Point, distIndex []int, k int) []Point

ComputeCentroids - Recalculates centroids position

func (Point) Norm

func (point Point) Norm() float64

Norm - returns the point norm

func (Point) PointDist

func (point Point) PointDist(p2 Point) float64

PointDist - returns the distance between to Points

func (Point) PointEqual

func (point Point) PointEqual(p2 Point) bool

PointEqual - Checks if a point is equal to another

func (Point) Print

func (point Point) Print()

Print - Prints the point vaues

func (Point) Subtract

func (point Point) Subtract(p2 Point) Point

Subtract - returns the value by value subtraction of two points