41b252d820
Adds two unofficial env variables so advanced users can experiment: 1. PHOTOPRISM_FACE_KIDS_DIST=0.6950 (range: 0.1-1.5, -1 to disable) 2. PHOTOPRISM_FACE_IGNORE_DIST=0.86 (range: 0.1-1.5, -1 to disable)
35 lines
592 B
Go
35 lines
592 B
Go
package clusters
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestKmeansClusterNumberMatches(t *testing.T) {
|
|
const (
|
|
C = 8
|
|
)
|
|
|
|
var (
|
|
f = "data/bus-stops.csv"
|
|
i = CsvImporter()
|
|
)
|
|
|
|
d, e := i.Import(f, 4, 5)
|
|
if e != nil {
|
|
t.Errorf("Error importing data: %s\n", e.Error())
|
|
}
|
|
|
|
c, e := KMeans(1000, C, EuclideanDist)
|
|
if e != nil {
|
|
t.Errorf("Error initializing kmeans clusterer: %s\n", e.Error())
|
|
}
|
|
|
|
if e = c.Learn(d); e != nil {
|
|
t.Errorf("Error learning data: %s\n", e.Error())
|
|
}
|
|
|
|
if len(c.Sizes()) != C {
|
|
t.Errorf("Number of clusters does not match: %d vs %d\n", len(c.Sizes()), C)
|
|
}
|
|
}
|