kmeans

Documentation

Index

• type KMeans
  • func NewKMeans(k, maxIterations int, trainingSet [][]float32) *KMeans
  • func (km *KMeans) Cluster() error
  • func (km *KMeans) Labels() []int

Types

type KMeans

type KMeans struct {
	Centroids [][]float32
	Output    io.Writer
	// contains filtered or unexported fields
}

KMeans structure for book keeping the KMeans computation

func NewKMeans

func NewKMeans(k, maxIterations int, trainingSet [][]float32) *KMeans

NewKMeans create a KMeans object to store the result of KMeans clustering

func (*KMeans) Cluster

func (km *KMeans) Cluster() error

Cluster Performs KMeans clustering

func (*KMeans) Labels

func (km *KMeans) Labels() []int

Labels : Return the cluster labels