Trait finalfusion::similarity::EmbeddingSimilarityBy

pub trait EmbeddingSimilarityBy {
    fn embedding_similarity_by<F>(
        &self, 
        query: ArrayView1<f32>, 
        limit: usize, 
        skip: &HashSet<&str>, 
        similarity: F
    ) -> Option<Vec<WordSimilarityResult>>
    where
        F: FnMut(ArrayView2<f32>, ArrayView1<f32>) -> Array1<f32>;
}

Trait for embedding similarity queries with a custom similarity function.

Required methods

fn embedding_similarity_by<F>(
    &self,
    query: ArrayView1<f32>,
    limit: usize,
    skip: &HashSet<&str>,
    similarity: F
) -> Option<Vec<WordSimilarityResult>> where
    F: FnMut(ArrayView2<f32>, ArrayView1<f32>) -> Array1<f32>, 

Find words that are similar to the query embedding using the given similarity function.

The similarity function should return, given the embeddings matrix and the query vector a vector of similarity scores. At most, limit results are returned.

Loading content...

Implementors

impl<V, S> EmbeddingSimilarityBy for Embeddings<V, S> where
    V: Vocab,
    S: StorageView, 

fn embedding_similarity_by<F>(
    &self,
    query: ArrayView1<f32>,
    limit: usize,
    skip: &HashSet<&str>,
    similarity: F
) -> Option<Vec<WordSimilarityResult>> where
    F: FnMut(ArrayView2<f32>, ArrayView1<f32>) -> Array1<f32>, 

Loading content...