testinghelpers

package
v1.35.21 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: May 21, 2026 License: BSD-3-Clause Imports: 21 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

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 DistanceWrapper added in v1.30.0

func DistanceWrapper(provider distancer.Provider) func(x, y []float32) float32

func MatchesInLists

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

func NewDummyStore added in v1.23.1

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

func NewDummyStoreFromFolder added in v1.30.19

func NewDummyStoreFromFolder(storeDir string, 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 added in v1.26.0

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)

func RecallAndLatency added in v1.30.0

func RecallAndLatency(ctx context.Context, queries [][]float32, k int, index VectorIndex, truths [][]uint64) (float32, float32)

Types

type DistanceFunction

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

type VectorIndex added in v1.30.0

type VectorIndex interface {
	SearchByVector(ctx context.Context, vector []float32, k int, allow helpers.AllowList) ([]uint64, []float32, error)
}

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL