testinghelpers

Documentation

Index

• func BruteForce(logger logrus.FieldLogger, vectors [][]float32, query []float32, k int, ...) ([]uint64, []float32)
• func BuildTruths(logger logrus.FieldLogger, queriesSize int, vectorsSize int, ...) [][]uint64
• func MatchesInLists(control []uint64, results []uint64) uint64
• func NewDummyStore(t testing.TB) *lsmkv.Store
• func Normalize(vectors [][]float32)
• func RandomVecs(size int, queriesSize int, dimensions int) ([][]float32, [][]float32)
• func RandomVecsFixedSeed(size int, queriesSize int, dimensions int) ([][]float32, [][]float32)
• func ReadQueries(queriesSize int) [][]float32
• func ReadSiftVecsFrom(path string, size int, dimensions int) [][]float32
• func ReadVecs(size int, queriesSize int, dimensions int, db string, path ...string) ([][]float32, [][]float32)
• type DistanceFunction

Functions

func BruteForce

func BruteForce(logger logrus.FieldLogger, vectors [][]float32, query []float32, k int, distance DistanceFunction) ([]uint64, []float32)

nil vectors are ignored, this allows for deleting vectors or supplying sparse sets where not every id has a vec

func BuildTruths

func BuildTruths(logger logrus.FieldLogger, queriesSize int, vectorsSize int, queries [][]float32, vectors [][]float32, k int, distance DistanceFunction, path ...string) [][]uint64

func MatchesInLists

func MatchesInLists(control []uint64, results []uint64) uint64

func NewDummyStore

func NewDummyStore(t testing.TB) *lsmkv.Store

func Normalize

func Normalize(vectors [][]float32)

func RandomVecs

func RandomVecs(size int, queriesSize int, dimensions int) ([][]float32, [][]float32)

func RandomVecsFixedSeed

func RandomVecsFixedSeed(size int, queriesSize int, dimensions int) ([][]float32, [][]float32)

func ReadQueries

func ReadQueries(queriesSize int) [][]float32

func ReadSiftVecsFrom

func ReadSiftVecsFrom(path string, size int, dimensions int) [][]float32

func ReadVecs

func ReadVecs(size int, queriesSize int, dimensions int, db string, path ...string) ([][]float32, [][]float32)

Types

type DistanceFunction

type DistanceFunction func([]float32, []float32) float32