Skip to content

Commit a6e261e

Browse files
committed
Changed minimum expected number of clusters for k-means
1 parent 5e81f77 commit a6e261e

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

errors.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,5 +6,5 @@ var (
66
ErrEmptySet = errors.New("Empty training set")
77
ErrNotTrained = errors.New("You need to train the algorithm first")
88
ErrZeroIterations = errors.New("Number of iterations cannot be less than 1")
9-
ErrZeroClusters = errors.New("Number of clusters cannot be less than 1")
9+
ErrOneCluster = errors.New("Number of clusters cannot be less than 2")
1010
)

kmeans.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,8 @@ func KmeansClusterer(iterations, clusters int, distance DistanceFunc, online ...
5454
return nil, ErrZeroIterations
5555
}
5656

57-
if clusters < 1 {
58-
return nil, ErrZeroClusters
57+
if clusters < 2 {
58+
return nil, ErrOneCluster
5959
}
6060

6161
var d DistanceFunc

0 commit comments

Comments
 (0)