search

Documentation

Index

• type Neighbor
• type Neighbors
  • func (neighbors Neighbors) Describe()
• type Searcher
  • func New(embs embedding.Embeddings) (*Searcher, error)
  • func (s *Searcher) Search(query embedding.Embedding, k int, ignoreWord ...string) (Neighbors, error)
  • func (s *Searcher) SearchInternal(word string, k int) (Neighbors, error)
  • func (s *Searcher) SearchVector(query []float64, k int) (Neighbors, error)

Types

type Neighbor

type Neighbor struct {
	Word       string
	Rank       uint
	Similarity float64
}

Neighbor stores the word with cosine similarity value on the target.

type Neighbors

type Neighbors []Neighbor

func (Neighbors) Describe

func (neighbors Neighbors) Describe()

type Searcher

type Searcher struct {
	Items embedding.Embeddings
	Dim   int
}

func New

func New(embs embedding.Embeddings) (*Searcher, error)

func (*Searcher) Search

func (s *Searcher) Search(query embedding.Embedding, k int, ignoreWord ...string) (Neighbors, error)

func (*Searcher) SearchInternal

func (s *Searcher) SearchInternal(word string, k int) (Neighbors, error)

func (*Searcher) SearchVector

func (s *Searcher) SearchVector(query []float64, k int) (Neighbors, error)